如何根据特定列的值从CSV文件中选择值

Question

以此csv为例：

[Col1，Col2

1，3，

0，5，

1,4

0,7

说我有一个列表，并且仅当同一行中Col1的值为1时，才想将Col2的值附加到列表中。结果列表应为[3,4]。

如何使用熊猫而不使用iterrows（）？

Answer 1

listt = []
for x, y in zip(df['Col1'], df['Col2']):
    if x == 1:
        listt.append(y)

Answer 2

假设您已经将csv读入名为df的数据框，则可以使用此方法：

wanted = df.query('Col1 == 1')['Col2'].values