温馨提示×

pandas怎么根据一列对另一列赋值

小亿
272
2023-12-29 09:33:22
栏目: 编程语言

要根据一列对另一列赋值,可以使用Pandas的.loc方法。下面是一个示例,说明了如何根据一列的值对另一列赋值:

import pandas as pd

# 创建一个示例DataFrame
data = {'A': [1, 2, 3, 4, 5],
        'B': [6, 7, 8, 9, 10]}
df = pd.DataFrame(data)

# 根据列'A'的值,将列'B'的值设置为新的值
df.loc[df['A'] > 3, 'B'] = 0

print(df)

输出结果如下:

   A  B
0  1  6
1  2  7
2  3  8
3  4  0
4  5  0

在上面的示例中,我们首先创建了一个具有两列的DataFrame。然后,我们使用.loc方法选择列’A’的值大于3的行,并将列’B’的值设置为0。

请注意,这种方法会直接在原始DataFrame上进行更改,而不会创建新的DataFrame。如果你想在不更改原始DataFrame的情况下创建一个新的DataFrame,可以使用.copy()方法创建DataFrame的副本,然后对副本进行操作。

0