首页 > 解决方案 > 为什么 X = np.array() 在不同的场合返回一个大小为 (N, 1) 和 (N,) 的数组?

问题描述

实际上,我有几个问题。我正在使用 Anaconda Spyder IDE 上的股票市场数据进行一个简单的机器学习项目。我正在通过以下代码从数据框的特定列创建一个 numpy 数组:

Y = np.array (dataframe ['column name'])

问题是,对于一个数据集,Y 的大小为 (N,),而对于另一个数据集,Y 的大小为 (N, 1)。数据集包含一些股票的股票市场价格。具体来说,当我从下载的 csv 文件(大约有 250 个数据行和 6 列)加载数据集时,Y 的大小是(N,)。但是当我使用pandas_datareader从雅虎导入数据时, Y 的大小是(N, 1)。我正在为 6 只股票导入 500 天的股票数据。(N, 1) size 导致我接下来进行的一些计算出现问题。

所以,我的问题是,为什么上面的代码部分为 Y 返回不同的数组大小?我可以以某种方式预先确定尺寸吗?

标签: pythonarraysnumpyspyder

解决方案


推荐阅读