dataframe 转化为list
时间: 2023-06-07 12:11:58 浏览: 617
将 DataFrame 转换为 List 的方法取决于您想要生成的 List 的形式。下面是几种可能的方法:
1. 将整个 DataFrame 转换为 List:使用 DataFrame 的 values 属性将 DataFrame 转换为 Numpy 数组,然后将 Numpy 数组转换为 List。例如:
```python
import pandas as pd
df = pd.DataFrame({'A': [1, 2], 'B': [3, 4]})
lst = df.values.tolist()
print(lst)
```
这将输出一个包含 DataFrame 所有数据的嵌套 List:`[[1, 3], [2, 4]]`。
2. 将 DataFrame 转换为 List of Dicts:使用 DataFrame 的 to_dict 方法将每行转换为一个字典,然后将字典添加到 List 中。例如:
```python
import pandas as pd
df = pd.DataFrame({'A': [1, 2], 'B': [3, 4]})
lst = []
for index, row in df.iterrows():
lst.append(row.to_dict())
print(lst)
```
这将输出一个包含 DataFrame 所有行的字典的 List:`[{'A': 1, 'B': 3}, {'A': 2, 'B': 4}]`。
3. 将 DataFrame 特定列转换为 List:使用 DataFrame 的 iloc 或 loc 方法选择特定列,然后将列转换为 List。例如:
```python
import pandas as pd
df = pd.DataFrame({'A': [1, 2], 'B': [3, 4]})
lst = df['A'].tolist()
print(lst)
```
这将输出一个包含 DataFrame 列 A 的数据的 List:`[1, 2]`。
希望这有助于回答您的问题。
阅读全文