温馨提示×

pandas条件筛选赋值的方法是什么

小亿
240
2023-12-16 01:17:25
栏目: 编程语言

Pandas中的条件筛选赋值可以通过两种方法实现:

  1. 使用.loc.iloc方法配合布尔条件进行筛选赋值。例如:
import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
df.loc[df['A'] > 2, 'B'] = 10

上述代码会将满足条件df['A'] > 2的行的B列赋值为10。

  1. 使用np.where()函数进行条件筛选赋值。这种方法可以在一行代码中实现条件筛选赋值。例如:
import pandas as pd
import numpy as np

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
df['B'] = np.where(df['A'] > 2, 10, df['B'])

上述代码会将满足条件df['A'] > 2的行的B列赋值为10,其他行的B列保持不变。

0