Python csv writer 在单行中组合两列

Question

我正在尝试在 csv 文件的第一列中写入 img_id（单个值），在 csv 文件的其余列中写入图像特征数组（f_descriptor_T）。 我写了以下代码：

path=/home/dir
listing = os.listdir(path)    

for file in listing:
    path_img_ext = os.path.join(path, file) #combine filename and extension
    data_read = feature_io.ReadFromFile(path_img_ext) #read single features of single image in each iteration 

    base=os.path.basename(file) #Extract image_id from its name
    os.path.splitext(file)
    img_id = os.path.splitext(base)[0]

    f_descriptor = data_read[2] #Extract particular array, from multiple arrays of features 

    feature_flat=f_descriptor.flatten() #Make flat array

    f_descriptor_T = np.array(feature_flat).T #Transpose feature array to convert col data to row form. 

    with open("output_0.csv", "a") as f:
        writer = csv.writer(f)
        writer.writerow([img_id])
        writer.writerow(f_descriptor_T)

不幸的是，输出在第一行显示img_id，在第二行显示f_descriptor_T，每次循环迭代，如下图所示：Output

但我想将两者（变量和数组）组合在单行中。

注意：我尝试使用 zip 和 concatenate 将两者结合起来，但失败了。

所需的输出应该是这样的：Desired Output不需要打印列标题，只需要在单行中打印 img_id 和 f_descriptor_T 数组。

Answer 1

根据您当前的代码，f_descriptor_T 出现在不同的行上，因为您在对其余数据的不同 writerow 调用中提供了 f_descriptor_T（对 writerow 的每次调用都会创建一个新行）。

请注意，writerow 函数将要插入的数据作为列表形式的单个参数。

如果您希望它们位于同一行，则需要在同一个调用中提供它们。要按照您在“所需输出”图像中显示的方式执行此操作，您需要为作者提供一个单一的平面列表。

由于您有两种不同的数据类型（img_id 是字符串，f_descriptor_T 包含数字），我建议通过创建一个包含您的 img_id 的数组来执行此操作，然后添加 f_descriptor_T 的内容给它。您需要将 f_descriptor_T 转换为普通的 Python 列表。对于转换，您可以使用list 函数。

这里有一组最小的示例来演示该行为。

你想像示例 4 那样做。它类似于示例 2，但包含一个如何将 numpy 数组转换为普通 List 的示例。

import csv
import numpy as np

#example 1
with open("output_different_rows.csv", "w") as f:
    writer = csv.writer(f)
    writer.writerow(['foo'])
    writer.writerow(['bar'])

#example 2
with open("output_same_row.csv", "w") as f:
    writer = csv.writer(f)
    writer.writerow(['foo'] + ['bar'])
#example 3
with open("output_same_row_with_array.csv", "w") as f:
    writer = csv.writer(f)
    writer.writerow([['foo', 'bar'], 'too'])

#example 4
numpy_array = np.arange(0,9)
listified_array = list(numpy_array)
with open("output_same_row_numpy.csv", "w") as f:
    writer = csv.writer(f)
    writer.writerow(['foo'] + listified_array)

Answer 2

正在尝试重现您的 csv 文件

In [131]: arr = np.arange(10) # a flat array
In [132]: arr = np.array(arr).T
In [133]: arr
Out[133]: array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

我从一个平面文件开始，但仍然有一个平面文件（一维数组上的.T 什么都不做）

In [135]: with open('out.csv','a') as f:
     ...:     writer = csv.writer(f)
     ...:     writer.writerow(['id'])
     ...:     writer.writerow(arr.T)
     ...:     
In [136]: cat out.csv
id
0,1,2,3,4,5,6,7,8,9

顾名思义，每次调用writerow 都会写入一行。

但是使用savetxt，我可以轻松地在整齐的列和行中编写一个二维数组：

In [137]: np.savetxt('out.csv', arr.reshape(5,2), delimiter=',')
In [138]: cat out.csv
0.000000000000000000e+00,1.000000000000000000e+00
2.000000000000000000e+00,3.000000000000000000e+00
4.000000000000000000e+00,5.000000000000000000e+00
6.000000000000000000e+00,7.000000000000000000e+00
8.000000000000000000e+00,9.000000000000000000e+00
In [139]: np.savetxt('out.csv', arr.reshape(5,2), delimiter=',', fmt='%5d')
In [140]: cat out.csv
    0,    1
    2,    3
    4,    5
    6,    7
    8,    9

混合文本和数字有点棘手。

只需将两个变量放在列表中，我就可以将它们写入一行：

In [143]: with open('out.csv','w') as f:
     ...:     writer = csv.writer(f)
     ...:     writer.writerow(['id', arr])
     ...: 
     ...:     
In [144]: cat out.csv
id,[0 1 2 3 4 5 6 7 8 9]

实际上已经完成了：

f.writeline('%s,%s'%(id, arr))

也就是说，将列表格式化为字符串并写入。