python - 行选择结合了 pandas 数据框中索引的条件和列的条件

Question

我想根据索引中的值与特定列中的值相结合从数据框中选择行:

df = pd.DataFrame([[0, 2, 3], [0, 4, 1], [0, 20, 30], [40, 20, 30]], 
                  index=[4, 5, 6, 7], columns=['A', 'B', 'C'])


    A   B   C
4   0   2   3
5   0   4   1
6   0  20  30
7  40  20  30

与

df.loc[df['A'] == 0, 'C'] = 99

我可以选择列 A = 0 的所有行并替换列 C 中的值99，但如何选择列 A = 0 的所有行和 index < 6 。换句话说，我想将对索引的选择与对列的选择结合起来。

Answer 1

您可以在 loc 语句中使用多个条件:

df.loc[(df.index < 6) & (df.A == 0), 'C'] = 99