如何将dataframe中一列的数据转换成数组
时间: 2023-02-06 21:43:25 浏览: 215
DataFrame 将某列数据转为数组的方法
5星 · 资源好评率100%
可以使用 `pandas` 库中的 `DataFrame` 类的 `to_numpy` 方法来将一列的数据转换成数组。
例如,假设你有一个名为 `df` 的 `DataFrame`,其中有一列名为 `'col_name'`,那么可以使用以下代码将该列转换为数组:
```
import pandas as pd
# 将 'col_name' 列转换为数组
col_array = df['col_name'].to_numpy()
```
也可以使用 `df.iloc[:, <column_index>]` 来获取一列的数据,例如:
```
# 假设 'col_name' 是第 2 列
col_array = df.iloc[:, 1].to_numpy()
```
还可以使用 `df.values` 将整个 `DataFrame` 转换为数组,例如:
```
# 将整个 DataFrame 转换为数组
df_array = df.values
```
阅读全文