python怎么把二维数据表某一列的转为数组
时间: 2025-01-21 07:06:22 浏览: 19
将Pandas DataFrame的一列转换为Numpy数组
为了实现这一目标,可以利用pandas.Series.to_numpy()
方法或是直接通过.values
属性获取指定列的数值并将其转化为numpy数组。这两种方式均能有效地完成从DataFrame单列到numpy数组的转变。
import pandas as pd
import numpy as np
# 创建示例DataFrame
data = {'A': [1, 2, 3],
'B': [4, 5, 6]}
df = pd.DataFrame(data)
# 方法一:使用to_numpy()函数
column_A_as_array_1 = df['A'].to_numpy()
# 方法二:访问.values属性
column_A_as_array_2 = df['A'].values
print(type(column_A_as_array_1)) # 输出:<class 'numpy.ndarray'>
print(column_A_as_array_1) # 输出:[1 2 3]
print(type(column_A_as_array_2)) # 输出:<class 'numpy.ndarray'>
print(column_A_as_array_2) # 输出:[1 2 3]
值得注意的是,在较新的版本中推荐使用to_numpy()
而非.values
因为前者提供了更清晰的行为定义以及更好的性能表现[^1]。
相关推荐

















