pandas提取第一列、第二列和最后一列重新组成一个新的pd,不知道列明,只知道列索引号,完整代码示例
时间: 2024-09-24 17:17:16 浏览: 39
pandas 的str列运算.md
在pandas库中,你可以通过列索引来提取特定的列。假设你有一个DataFrame `df`,并且你想提取第一列(索引为0)、第二列(索引为1),以及最后一列(通常可以获取到列名或者索引为-1),你可以这样做:
```python
import pandas as pd
# 假设 df 是你的 DataFrame
first_col = df.iloc[:, 0]
second_col = df.iloc[:, 1]
# 如果最后一列是基于名称的,可以这样获取(如果列名是 'last_column')
last_col_name = 'last_column'
if last_col_name in df.columns:
last_col = df[last_col_name]
else:
# 如果是基于索引的未知位置,这里默认索引 -1 表示最后一列
last_col = df.iloc[:, -1]
# 将它们组合成新的 DataFrame
new_df = pd.concat([first_col, second_col, last_col], axis=1)
```
请注意,上述代码假设你知道最后一列的名称或索引。如果列名不确定,你需要遍历列名或者使用 `df.columns[-1]` 来尝试获取最后一列。
阅读全文