Как выбрать значения из CSV-файла в зависимости от значения указанного столбца c

Question

Возьмите этот CSV в качестве примера:

Col1, Col2
1,3,
0,5,
1,4
0,7

Скажите, что у меня есть список, и я хочу добавить значения Col2 в список, только если значение Col1 в такая же строка 1. В результате должен получиться список [3,4].

Как я могу сделать это с помощью pandas без использования iterrows()?

Answer 1

1. Я думаю, что .loc - это самый pythoni c и самый простой метод.

   df.loc[df['Col1'] == '1']['Col2'].to_list()
   

2. Альтернатива с использованием np.where:

   import numpy as np
   df['Col3'] = np.where(df['Col1'] == 1, df['Col2'], '')
   listt = list(filter(None,df['Col3'].to_list()))
   

3. zip - это al oop альтернатива iterrows.

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

Answer 2

import pandas as pd

lst_result=[]
for ind, row in df.iterrows():
    if df.at[ind,'Col1'] == 1:
        lst_result.append(df.at(ind,['Col2']))
print(lst_result)

Answer 3

Предполагая, что вы прочитали в своем csv фрейм данных с именем df, вы можете использовать это:

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