通过index 查询dataframe的元素并组成新的dataframe
时间: 2023-08-20 13:01:56 浏览: 82
通过 index 查询 DataFrame 的元素可以使用 `loc` 方法,并将查询结果组成新的 DataFrame。
例如,假设存在如下的 DataFrame:
```python
import pandas as pd
data = {'A': [1, 2, 3, 4],
'B': [5, 6, 7, 8],
'C': [9, 10, 11, 12]}
df = pd.DataFrame(data, index=['row1', 'row2', 'row3', 'row4'])
```
可以通过 index 查询元素并组成新的 DataFrame,具体步骤如下:
```python
# 查询 index 为 'row2' 的元素,并组成新的 DataFrame
new_df = df.loc[['row2']]
```
以上代码会生成一个新的 DataFrame `new_df`,其中包含原 DataFrame 中 index 为 'row2' 的行对应的元素。
如果想要同时查询多个 index 对应的元素,并组成新的 DataFrame,可以将这些 index 放在一个列表中进行查询:
```python
# 查询 index 为 'row2' 和 'row4' 的元素,并组成新的 DataFrame
new_df = df.loc[['row2', 'row4']]
```
以上代码会生成一个新的 DataFrame `new_df`,其中包含原 DataFrame 中 index 为 'row2' 和 'row4' 的行对应的元素。
通过以上方法,可以根据 index 查询 DataFrame 的元素,并将查询结果组成新的 DataFrame。
阅读全文