Question

我是python和pandas的完全新手。我想迭代数据框中的所有行并检查“Class”列中的元素是否为1？怎么做到这一点？

另外，我想将这些特定行追加到数据帧中？喜欢这个

         emptydataframe = pd.DataFrame(columns=['A','B','C','D','E','F','G'])
         data = pd.read_csv('data/trainData.csv')
         count = 0
         for rows in data:
            if(data[rows]["Class"] == 1):
                 count+= 1
                 emptydataframe.append(data[rows])

我该怎么做？

Answer 1

如果我理解正确 - 您不想循环使用DF：

In [185]: df
Out[185]:
    A   B   C  Class
0   1   2   3      0
1   4   5   6      1
2   7   8   9      1
3  10  11  12      0

In [186]: new = df.loc[df['Class']==1]

In [187]: new
Out[187]:
   A  B  C  Class
1  4  5  6      1
2  7  8  9      1

如果和追加条件pandas dataframe？

1 个答案: