将numpy.ndarray写入xlsx电子表格

时间:2018-08-18 00:00:11

标签: python arrays python-2.7 numpy xlsxwriter

我正在尝试使用xlsxwriter将numpy.ndarray写入Excel电子表格。

我的数组的“类型”是numpy.ndarray。数组本身在打印时采用以下形式:

exampleArray = array([[[0.    ,  1], [0.    , 1], [0.    ,1]]])

我正在使用的代码如下:

workbook = xlsxwriter.Workbook('WorkBookName.xlsx')
worksheet = workbook.add_worksheet('workSheetName')

row = 0
col = 0

for data in exampleArray[0]:
    worksheet.write_colum(row, col data)
    col += 1

workbook.close()

我得到了错误:

AttributeError:“列表”对象没有属性“告诉”

谢谢!

1 个答案:

答案 0 :(得分:3)

使用熊猫,这很简单。您只需要确保数组是二维的即可。

import pandas as pd

A = np.array([[[0.    ,  1], [0.    , 1], [0.    ,1]]])

# construct dataframe, index [0] to make 2d
df = pd.DataFrame(A[0])

# save to Excel, exclude index and headers
df.to_excel('file.xlsx', index=False, header=False)