pandas.np.where
时间: 2023-11-12 16:57:05 浏览: 306
pandas.np.where是pandas和numpy库中的一个函数,用于根据指定的条件对数组或DataFrame对象的元素进行逐个判断,并返回一个新的数组或DataFrame对象,其中满足条件的元素被替换为指定的值,不满足条件的元素则保留原值。
具体来说,pandas.np.where函数的语法如下:
```python
pandas.np.where(condition, x, y)
```
其中,condition为一个布尔型数组或条件表达式,x和y分别为两个数组或标量,用来指定替换满足条件的元素和不满足条件的元素的值。如果x和y都是标量,则所有满足条件的元素都将替换为x;如果x和y都是数组,则需要保证它们的形状相同,并且满足条件的元素将替换为x中对应位置的元素,不满足条件的元素将替换为y中对应位置的元素。
下面是一个简单的示例:
```python
import pandas as pd
import numpy as np
data = {'A': [1, 2, 3, 4, 5], 'B': [6, 7, 8, 9, 10]}
df = pd.DataFrame(data)
df['A'] = np.where(df['A'] > 3, -1, df['A'])
print(df)
```
输出结果为:
```
A B
1 6
1 2 7
2 3 8
3 -1 9
4 -1 10
```
在上面的示例中,我们使用np.where函数对DataFrame对象的列'A'进行了条件判断和替换操作,将所有大于3的元素替换为-1。
阅读全文