在python中搜索csv文件中的值列表

Question

我有一份清单如下。

[[0, 0.1, 0.3], [0.5, 0.2, 0.8]]

我还有一个csv文件如下。

No, heading1, heading2, heading3
0, 0, 0.7, 0.3
1, 0, 0.1, 0.3

现在我想只使用'heading1'，'heading2'和'heading3'中的值搜索这些值列表，如果匹配则返回与其对应的'No'。

我们可以使用熊猫吗？

Answer 1

您可以在所有列中使用merge：

L = [[0, 0.1, 0.3], [0.5, 0.2, 0.8]]

#helper df with same columns without first
df1 = pd.DataFrame(L, columns= df.columns[1:])
print (df1)
   heading1  heading2  heading3
0       0.0       0.1       0.3
1       0.5       0.2       0.8

#after merge select column No
d = pd.merge(df, df1)['No']
print (d)
0    1
Name: No, dtype: int64