python - 在这种情况下如何使用 np.stack？

Question

我有两个 np.ndarray :

predictions = np.array([[0.2, 0.9], [0.01, 0.0], [0.3, 0.8], ...])
filenames = np.array(["file1", "file2", "file3", ...])

文件名中的每个文件对应于预测中的每个数组:

文件1==>[0.2, 0.9]

文件2==>[0.01, 0.0]

文件3==>[0.3,0.8]...

我想将这两个数组中的值打印成一个csv文件，如下所示:

fileName        label1      label2
file1           0.2         0.9 
file2           0.1         0.0
file3           0.3         0.8

我希望用np.stack把这两个np.array合并成一个数据结构，然后用np.savetext(path, array, )输出到csv文件。

但 np.stack(array, axis=1) 似乎只接受两个具有相同形状的数组。有没有办法让堆栈适用于这种情况？

Answer 1

使用 numpy.expand_dims 的解决方案和 numpy.hstask套路:

import numpy as np
result = np.hstack((np.expand_dims(filenames, axis=1), predictions))

# saving to csv file using `np.savetxt`:
with open('./text_files/predictions.csv', 'wb') as fh:
    np.savetxt(fh, X= result, header='fileName\tlabel1\tlabel2', delimiter='\t', fmt='%-8s\t%-6s\t%-6s')

predictions.csv(测试文件)内容:

# fileName  label1  label2
file1       0.2     0.9   
file2       0.01    0.0   
file3       0.3     0.8