Excel 到 pandas 到 numpy 数组的转换

Question

我可以使用 pandas 将 excel 数据导入 python 3.6。

我用于此目的的代码如下：

import pandas as pd

import numpy as np

df=pd.read_excel("I:/Python/Excel.xlsx")

df.head()

上面的代码显示了 Jupyter notebook 上的 Table1。

我不知道如何将我可以在 Jupyter 笔记本中看到的 Table1（从上面的代码中获得）中的两列转换为 numpy 数组，以便对它们进行进一步分析。

此表共有 17 列。我需要将 pandas 框架中的 'Column 15' 和 'Column 16' 工作到 numpy 数组中，以便最终结构如下：

data_set=[[x1,y1],[x2,y2],...[x1000,y1000]]

# x1 = row1 from Column 15
# y1 = row1 from Column 16

我最终将使用这些数据将“xy 散点图”绘制到 matplotlib 中。这部分我知道该怎么做。从这个论坛学到了很多东西。谢谢大家！

我在网上寻找这个解决方案，但找不到解决方案。

感谢阅读。

Answer 1

我认为需要按iloc 的位置选择列，转换为 numpy 数组，如果需要列出：

np.random.seed(1245)

df = pd.DataFrame(np.random.randint(10, size=(3, 17))).add_prefix('data')
print (df)
   data0  data1  data2  data3  data4  data5  data6  data7  data8  data9  \
0      6      5      3      5      0      6      9      3      4      6   
1      7      1      2      9      3      9      8      8      7      1   
2      4      1      8      2      5      6      6      5      1      5   

   data10  data11  data12  data13  data14  data15  data16  
0       9       4       9       5       2       7       9  
1       7       5       5       2       2       4       0  
2       4       6       6       0       2       7       6  

data_set = df.iloc[:, [14, 15]].values.tolist()
#alternative
#data_set = df.values[:, [14, 15]].tolist()
print(data_set)
[[2, 7], [2, 4], [2, 7]]

编辑：

#if want select columns by names
x = df['data01']
y = df['data02']

#if want select columns by positions
#x = df.iloc[:, 1]
#y = df.iloc[:, 2]

s = plt.scatter(x, y)

Answer 2

你不能用x=np.array(df)吗？ 然后x 将成为您的新数组。