dataframe 某行某几列均相同
时间: 2024-05-26 07:07:44 浏览: 86
如果您想要判断DataFrame中的某一行中的某几列的值是否全部相同,可以使用`nunique`函数来实现。该函数可以返回该行中独特值的数量。如果该行中的某几列值全部相同,则`nunique`函数的返回值应为1。以下是一个示例代码:
```python
import pandas as pd
# 创建DataFrame
df = pd.DataFrame({'A': [1, 2, 3, 4], 'B': [1, 1, 2, 2], 'C': [1, 1, 1, 1]})
# 判断第三行中的'A'和'B'两列的值是否全部相同
if df.loc[2, ['A', 'B']].nunique() == 1:
print("第三行中'A'和'B'两列的值全部相同")
else:
print("第三行中'A'和'B'两列的值不全部相同")
```
如果您需要判断DataFrame中的所有行中的某几列的值是否全部相同,可以使用`apply`函数结合`nunique`函数来实现。以下是示例代码:
```python
import pandas as pd
# 创建DataFrame
df = pd.DataFrame({'A': [1, 2, 3, 4], 'B': [1, 1, 2, 2], 'C': [1, 1, 1, 1]})
# 判断所有行中的'A'和'B'两列的值是否全部相同
result = df[['A', 'B']].apply(lambda x: x.nunique() == 1, axis=1)
print(result)
```
这个示例代码会输出一个`Series`对象,其中每个元素代表对应行中的'A'和'B'两列的值是否全部相同。
阅读全文