python - 通过从 Excel 读取数据创建矩阵

Question

我正在尝试使用 xlrd 从 XML 文件创建数据矩阵。下面是 XML 数据和我的尝试:

import xlrd
file_loc="C:\\Users\\xxxx\\Documents\\test1.xlsx"
wkb=xlrd.open_workbook(file_loc)
sheet=wkb.sheet_by_index(0)

_maxtrix=[]
for col in range (sheet.ncols):
    for row in range (sheet.nrows):
        _matrix.append(sheet.cell_value(row,col))
print _matrix

我的输出:

[1.0, 2.0, 3.0, 4.0, 2.0, 3.0, 4.0, 5.0, 3.0, 4.0, 5.0, 6.0, 4.0, 5.0, 6.0, 7.0]

期望的输出:

[[1,2,3,4],
[2,3,4,5],
[3,4,5,6],
[4,5,6,7]]

关于如何获得所需输出的任何想法？

Answer 1

另一种可能的方法是使用 pandas。

您可以使用方法read_excel读取您的csv，然后将其转换为矩阵。

这里是一个例子:

import pandas as pd
df = pd.read_excel("something.xlsx", sheet_name=0)
a = df.as_matrix()