pandas 列数据转换成数组|极客教程

pandas 列数据转换成数组

在数据处理和分析中，pandas 是一个非常常用的库，它提供了丰富的数据结构和数据处理工具，其中最常见的数据结构是 Series 和 DataFrame。有时候我们需要将 pandas 的列数据转换成数组，这样可以更方便地进行一些操作和分析。本文将详细介绍如何将 pandas 列数据转换成数组。

1. 将 Series 转换成数组

我们首先来看一下如何将 pandas 的 Series 转换成数组。假设我们有如下的一个 Series：

import pandas as pd

data = {'A': [1, 2, 3, 4, 5],
        'B': ['geek-docs.com', 'hello', 'world', 'geek-docs.com', 'python']}
df = pd.DataFrame(data)
s = df['B']

print(s)

运行结果：

0    geek-docs.com
1           hello
2           world
3    geek-docs.com
4          python
Name: B, dtype: object

现在我们要将 Series s 转换成数组，可以使用 to_numpy() 方法：

arr = s.to_numpy()
print(arr)

运行结果：

['geek-docs.com' 'hello' 'world' 'geek-docs.com' 'python']

通过 to_numpy() 方法，我们将 Series 转换成了一个数组。这样我们就可以更方便地对数据进行操作了。

2. 将 DataFrame 中的某一列数据转换成数组

除了将 Series 转换成数组，有时候我们也需要将 DataFrame 中中的某一列数据转换成数组。同样是借助 to_numpy() 方法，我们可以很容易地实现这个功能。下面是一个示例代码：

import pandas as pd

data = {'A': [1, 2, 3, 4, 5],
        'B': ['geek-docs.com', 'hello', 'world', 'geek-docs.com', 'python']}
df = pd.DataFrame(data)

print(df)

arr = df['B'].to_numpy()
print(arr)

运行结果：

   A              B
0  1  geek-docs.com
1  2          hello
2  3          world
3  4  geek-docs.com
4  5         python

['geek-docs.com' 'hello' 'world' 'geek-docs.com' 'python']

如上所示，我们可以轻松地将 DataFrame 中的某一列数据转换成数组。这在数据处理和分析中非常有用。