python - 将具有不同大小的单个数组的二维 numpy 字符串数组保存到 csv 文件

Question

我有一个 2D numpy 字符串数组。我正在尝试将其保存到 CSV 文件。因此，当一维数组的大小不同时，就会出现问题。

例如:

b = [['a','b']    #size of single array = 2
     ['c','d']]   #size of single array = 2

所以，现在如果我尝试使用以下方法保存它:

np.savetxt("filename.csv", b, fmt ="%s", delimiter=",")

输出 csv 文件为:

Col1      Col2     
-------    ----     
a          b        
c          d          

这就是我想要的，但现在假设我有一个不同大小的一维数组。

例如:

b = [['a','b']    #size of single array = 2
     ['c']]       #size of single array = 1

现在，当我尝试保存它时，输出文件是:

Col1      Col2     
-------   ----     
['a'      'b']           
['c']                   

而我希望将其另存为:

Col1       Col2     
-------    ----     
a          b        
c                   

有人可以帮我吗？

Answer 1

您可以先将其转换为 pandas 数据框，然后保存为 csv:

import pandas as pd

b = [['a','b'], ['c']]
df = pd.DataFrame(b)
df.fillna('', inplace=True)
df.to_csv(path)

但是您询问了 numpy 数组。如果你有 numpy 列表数组，那么你可以先将其转换为列表列表:

import numpy as np
import pandas as pd

b = np.array([['a','b'], ['c']])
b = list(b)
df = pd.DataFrame(b)
df.fillna('', inplace=True)
df.to_csv(path)