欢迎访问宙启技术站
智能推送

使用to_array()方法将Pandas中的数据框转换为数组(Python示例)

发布时间:2024-01-02 14:24:42

在Pandas中可以使用to_numpy()方法将数据框转换为数组。下面是一个使用例子:

首先,我们需要安装和导入Pandas库:

!pip install pandas
import pandas as pd

然后,我们可以创建一个包含一些数据的数据框:

data = {'Name': ['Tom', 'Nick', 'John', 'Lucy'],
        'Age': [25, 30, 18, 22],
        'City': ['New York', 'Paris', 'London', 'Tokyo']}
df = pd.DataFrame(data)

现在,我们可以使用to_numpy()方法将数据框转换为数组:

array = df.to_numpy()
print(array)

输出结果为:

[['Tom' 25 'New York']
 ['Nick' 30 'Paris']
 ['John' 18 'London']
 ['Lucy' 22 'Tokyo']]

注意,to_numpy()方法会将数据框的所有列转换为数组的行,每一列的元素类型会根据数据框列的类型进行自动转换。在上面的例子中,由于NameCity列是字符串类型,而Age列是整数类型,所以转换后的数组中的元素都是对象类型。

需要注意的是,使用to_numpy()方法转换为数组后会丢失数据框的列索引和行索引信息,仅仅保留了数值。

同时,也可以使用values属性来实现类似的功能,即将数据框转换为一个Numpy数组:

array = df.values
print(array)

这样也会得到与上面相同的输出结果。

总结:在Pandas中,可以使用to_numpy()方法或values属性将数据框转换为Numpy数组。这些方法非常简单而且高效,可以方便地将数据框中的数据转换为数组进行进一步的处理和分析。