pandas选取索引列中数值相同的某几行
时间: 2023-07-09 10:50:53 浏览: 112
Python Pandas 获取列匹配特定值的行的索引问题
你可以使用 Pandas 的 `duplicated()` 和 `drop_duplicates()` 函数来选取索引列中数值相同的某几行。
举个例子,如果你的 DataFrame 如下所示:
```
import pandas as pd
df = pd.DataFrame({
'id': [1, 2, 3, 3, 4, 4, 5],
'value': [10, 20, 30, 30, 40, 40, 50]
})
```
其中 `id` 列是索引列,你可以使用以下代码选取 `id` 列中数值为 3 和 4 的行:
```
df = df.loc[df.index.duplicated(keep=False) & df['id'].isin([3, 4])]
df = df.drop_duplicates(subset='id')
```
第一行代码中,`df.index.duplicated(keep=False)` 会返回所有重复的索引行,`df['id'].isin([3, 4])` 会返回 `id` 列中数值为 3 和 4 的行。使用 `&` 运算符将两个条件合并,选出 `id` 列中数值为 3 和 4 并且索引列中重复的行。
第二行代码中,`df.drop_duplicates(subset='id')` 会根据 `id` 列去重,保留第一个出现的行。这样就能得到 `id` 列中数值为 3 和 4 的第一行。
阅读全文