Question

我有一个数据框，想添加一列新的bool值，引用行号列表。

>>> df
    col1  col2
0     1     1
1     2     2
2     4     3
3     8     4

>>> lst_rowNumbers
[1, 3]

结果将如下所示：

    col1  col2   bool
0     1     1  False
1     2     2   True
2     4     3  False
3     8     4   True

我以为这可以用，但是没有用。

>>> df['bool'] = False
>>> df.iloc[ lst_rowNumbers ]['bool'] = True

我该如何处理熊猫？

Answer 1

如果要按索引名称选择：

assignableAreas

或者：

df['bool'] = False
df.loc[ lst_rowNumbers , 'bool'] = True

df['bool'] = df.index.isin(lst_rowNumbers)

如果需要按位置选择，请通过Index.get_loc获取列名的位置：

print (df)
   col1  col2   bool
0     1     1  False
1     2     2   True
2     4     3  False
3     8     4   True

或将isin与通过索引返回的实际索引值一起使用：

print (df)
   col1  col2
a     1     1
b     2     2
c     4     3
d     8     4

lst_rowNumbers = [1,3]
df['bool'] = False
df.iloc[ lst_rowNumbers , df.columns.get_loc('bool')] = True

df['bool'] = df.index.isin(df.index[lst_rowNumbers])

如何在具有行号列表的数据框中添加一列？

1 个答案: