用Python从list中提取3列,将3列中第一行元素设为列索引
时间: 2024-05-09 19:17:00 浏览: 112
Python 从列表中取值和取索引的方法
可以使用 Pandas 库来完成这个任务,具体代码如下:
```python
import pandas as pd
data = [['A', 'B', 'C'], [1, 2, 3], [4, 5, 6], [7, 8, 9]]
df = pd.DataFrame(data[1:], columns=data[0])
# 从 DataFrame 中提取第一列、第二列和第三列
column1 = df.iloc[:, 0]
column2 = df.iloc[:, 1]
column3 = df.iloc[:, 2]
# 将第一行元素设为列索引
df.columns = list(df.iloc[0])
df = df.drop(0)
print(df)
```
其中,将 `data` 列表转换成 `DataFrame`,然后使用 `iloc` 方法从中提取第一列、第二列和第三列,最后将第一行元素设为列索引即可。注意要将原始数据列表中的第一行元素舍去,因为它已经被用作了列索引。
阅读全文