在 csv 中写入多个 numpy 数组以及字符串

Question

看起来很简单，但我失败了。我想将以下数组与其他字符串一起写入 csv 文件中的一行（即"\n"）。值应以分号分隔。

我从列表中创建数组

auth_list = ['5,2.6,5.23515389636e-11,0.00444005161574,0.0644299145707,0.04,0.0037037037037', '1,5.0,5.65187364062e-12,0.0,0.0605326876513,0.0,0.000740740740741']
aff_list = [(1, 0.003105590062111801), (1, 0.001838235294117647)]
comm_list = ['', '4,17.5,0.00584085856718,0.0002890919969,0.278790504782,0.0140752484561,0.0029973357016,0.00044404973357', '0,0.0,0.000218693005322,3.33471210416e-07,0.232075228649,0.0,0.000222024866785,0.0', '0,0.0,0.00025235732634,6.73003774237e-07,0.233652374653,0.0428571428571,0.000555062166963,0.0']

我继续把它们变成数组

import numpy
auth_array = numpy.genfromtxt(auth_list, delimiter=",")
aff_array = numpy.array(aff_list) # Here I don't have to use numpy.genfromtext, don't know why
auth_array = numpy.genfromtxt(aff_list, delimiter=",")

这些数组已经没有逗号。我使用计算方法

auth_mean = numpy.mean(auth_array, axis=0)
aff_mean = numpy.mean(aff_array, axis=0)
comm_mean = numpy.mean(comm_array, axis=0)
title = "Happy New Year"

使用print，我在终端看到

auth_mean = [  3.00000000e+00   3.80000000e+00   2.90017063e-11   2.22002581e-03
          6.24813011e-02   2.00000000e-02   2.22222222e-03]
aff_mean = [ 1.          0.00247191]
comm_mean = [  1.33333333e+00   5.83333333e+00   2.10396963e-03   9.66994906e-05
          2.48172703e-01   1.89774638e-02   1.25814091e-03   1.48016578e-04]

数组总是具有相同的维度。

output_text = title + str(auth_mean).strip('[]') + ";" + str(com_mean).strip('[]') + ";" + str(aff_mean).strip('[]') + "\n"
output_file = open(output_file_name, 'w')
output_file.write(output_text)
output_file.close()

产量

  Happy New Year;3.00000000e+00   3.80000000e+00   2.90017063e-11   2.22002581e-03
  6.24813011e-02   2.00000000e-02   2.22222222e-03;  1.33333333e+00   5.83333333e+00   2.10396963e-03   9.66994906e-05
  2.48172703e-01   1.89774638e-02   1.25814091e-03   1.48016578e-04; 1.          0.00247191

我怎样才能画出一条简单的直线

Happy New Year;3.00000000e+00;3.80000000e+00;2.90017063e-11;2.22002581e-03;6.24813011e-02;2.00000000e-02;2.22222222e-03;1.33333333e+00;5.83333333e+00;2.10396963e-03;9.66994906e-05;2.48172703e-01;1.89774638e-02;1.25814091e-03;1.48016578e-04;1.;0.00247191

Answer 1

我添加了逗号来分隔您的三个列表的元素（如上面 MERose 所查询的），并且我还必须初始化 output_file_name：output_file_name = "output.csv"。

我希望这不是一个愚蠢的问题，但是您是否可以在打开“自动换行”的文本编辑器中检查输出文件？我运行了代码，我得到了你需要的一行输出。如果我随后在打开“自动换行”的情况下在 Notepad++ 中查看该文件，它似乎位于多行，而实际上它只是一行。

仅供参考：输出的文件也可以很好地导入 LibreOffice Calc。

Answer 2

首先，您在代码中忘记了逗号，我认为是：

title = "Happy New Year"
auth = [  3.00000000e+00,   3.80000000e+00,   2.90017063e-11,   2.22002581e-03,
          6.24813011e-02,   2.00000000e-02,   2.22222222e-03]
aff = [ 1.  ,        0.00247191]
comm = [  1.33333333e+00,   5.83333333e+00,   2.10396963e-03,   9.66994906e-05,
          2.48172703e-01,   1.89774638e-02,   1.25814091e-03,   1.48016578e-04]

现在关于问题： 您可以使用itertools chain 将多个数组合并为一个：

from itertools import chain

out = chain([title], auth, aff, comm, ['\n'])

那你可以用分号加入这一行：

str_row = ';'.join(map(str, out))
print(str_row)  #or write to file, if you need

但是，我强烈建议您不要重新发明轮子并使用标准 csv module 来编写任何 csv 文件。您可以在设置 csv writer 时指定';' 作为分隔符。