【问题标题】:Convert pandas DataFrame columns to NumPy array (extension question)将 pandas DataFrame 列转换为 NumPy 数组(扩展问题)
【发布时间】:2021-05-29 16:45:59
【问题描述】:

我的问题与 cs95 在这篇文章中的出色回答有关:

Convert pandas dataframe to NumPy array

它提供了一种将 pandas DataFrame 转换为 numpy-matrix 的快速简便的方法。希望您原谅我直接引用他的设置:

df = pd.DataFrame(data={'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]}, 
                  index=['a', 'b', 'c'])

# Convert the entire DataFrame
df.to_numpy()
# array([[1, 4, 7],
#        [2, 5, 8],
#        [3, 6, 9]])

# Convert specific columns
df[['A', 'C']].to_numpy()
# array([[1, 7],
#        [2, 8],
#        [3, 9]])

我需要将 DataFrames columns 转换为 numpy-array。在他的回答中,我们得到的是一个带有 rows 的数组。

我希望能够得到这样的东西

columns = df.other_to_numpy()
#array([[1,2,3],
#       [4,5,6],
#       [7,8,9]])

因为这允许我将原始列作为columns[0] 引用到columns[2]。我意识到这个问题与我所引用的帖子非常接近,但在对上述帖子的回答中,我找不到这个问题变体的答案。如果这是一个非常简单的问题,请您耐心等待。

【问题讨论】:

  • 简单地转置数据帧。试试df.T.to_numpy()df.to_numpy().T

标签: python arrays pandas dataframe numpy


【解决方案1】:

你的意思是你想要转置数组?

df.T.values

【讨论】:

    猜你喜欢
    • 2019-08-04
    • 2020-02-10
    • 2018-12-27
    • 1970-01-01
    • 1970-01-01
    • 2020-08-20
    • 1970-01-01
    • 2019-05-05
    • 2015-10-25
    相关资源
    最近更新 更多