如果和附加条件Pandas数据帧？

Question

我是python和pandas的新手，我想遍历dataframe中的所有行，检查“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

如果我理解正确-你不想循环查看你的数据框：

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