[人工智能]DataFrame删除指定条件的行

import pandas as pd

df1 = pd.DataFrame(data=[{'symbol': '000001', 'close': 1.0}, {'symbol': '000002', 'close': 2.0}, {'symbol': '000001', 'close': 3.0}])
symbols = ['000001', '000002']
print(df1.index[df1['symbol'] == '000001'])
print(df1.index[(df1['symbol'] == '000001') & (df1['close'] == 1.0)])

df1.drop(df1.index[(df1['symbol'] == '000001')], inplace=True)
print(df1)

使用df.index函数获取条件索引。

参考链接：

(8 封私信) 在 Pandas 中如何获取列与特定值匹配的行的索引？ - 知乎 (zhihu.com)