我需要获取给定列中包含特定值的行号.可以保证这样的值存在于列和唯一行中.
我的try 是:
import pandas as pd
pos = pd.DataFrame(columns=['id', 'pred'])
pos.loc[1,'id'] = [4, 4, 4]
pos.loc[2,'id'] = [2, 3, 3]
pos.loc[3,'id'] = [1, 1, 2]
pos.loc[4,'id'] = [1, 2, 1]
print(pos)
print(pos[pos['id'] == [1, 1, 2]].index[0])
我收到一个错误:
ValueError: ('Lengths must match to compare', (6,), (3,))