pandas判断两个df数据相等
时间: 2023-08-24 07:19:41 浏览: 39
pandas数据选取:df[] df.loc[] df.iloc[] df.ix[] df.at[] df.iat[]
5星 · 资源好评率100%
你可以使用 `pandas.testing.assert_frame_equal()` 函数来判断两个 DataFrame 是否相等。这个函数会比较 DataFrame 中每一个元素是否相等,如果有任何一个元素不相等,函数会抛出 AssertionError。下面是一个例子:
```python
import pandas as pd
from pandas.testing import assert_frame_equal
df1 = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
df2 = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
assert_frame_equal(df1, df2) # 没有抛出异常,说明两个 DataFrame 相等
```
如果你想忽略 DataFrame 中的一些列来进行比较,可以使用 `check_like` 参数。例如,下面的代码只比较 `df1` 和 `df2` 中的 `A` 列:
```python
assert_frame_equal(df1[['A']], df2[['A']], check_like=True)
```
阅读全文