dataframe某一列转为数组
时间: 2023-09-08 21:08:26 浏览: 64
可以使用 pandas 库中的 values 属性将 dataframe 某一列转为数组。
例如,假设我们有以下的 dataframe:
```python
import pandas as pd
df = pd.DataFrame({'a': [1, 2, 3], 'b': [4, 5, 6]})
```
我们可以通过以下语句,将 a 列转换为 numpy 数组:
```python
a_array = df['a'].values
```
同样地,我们也可以将 b 列转换为 numpy 数组:
```python
b_array = df['b'].values
```
相关问题
统计dataframe的频数并转为数组
您可以使用pandas库中的value_counts()函数来统计DataFrame中每个元素的频数,并将结果转换为数组。以下是一个示例代码:
```python
import pandas as pd
# 创建一个示例DataFrame
df = pd.DataFrame({'A': ['apple', 'banana', 'apple', 'banana', 'banana'],
'B': ['cat', 'dog', 'cat', 'dog', 'dog']})
# 统计'A'列元素的频数并转为数组
freq_array = df['A'].value_counts().values
print(freq_array)
```
运行上述代码,将得到输出结果:
```
[3 2]
```
这表示在'A'列中,'apple'出现了3次,'banana'出现了2次。频数被转换成了一个数组。
dataframe中某一列转为一维数组
可以使用 pandas 库中的 .to_numpy() 方法将 dataframe 中的某一列转换为一维数组。例如,如果要将 dataframe 中的列名为 'column_name' 的列转换为一维数组,可以使用以下代码:
```python
import pandas as pd
# create a sample dataframe
df = pd.DataFrame({'column_name': [1, 2, 3, 4, 5]})
# convert the column to a one-dimensional array
array = df['column_name'].to_numpy()
```
在这个例子中,我们首先创建了一个名为 `df` 的 dataframe,其中包含一个名为 'column_name' 的列。接下来,我们使用 .to_numpy() 方法将 'column_name' 列转换为一维数组,并将结果存储在名为 'array' 的变量中。