从CSV文件写入CSV文件

Question

有一个问题。如果我运行这个代码，我就可以得到CSV中所包含的内容。我的目标是获取一个文本/data/csv文件，它看起来如下：“红色”、“绿色”、“蓝色”，意思是：

1. 将单个列转换为一行。
2. 在转换为行时，输入逗号以区分值。

可以通过Python完成吗？有什么网上资料我可以查吗？

Answer 1

csv文件是按顺序读取的，也就是说，每次只返回一行数据。但是，在读取文件并丢弃其余数据时，可以只生成所需值的数组。

import csv

with open('example.csv') as csvfile:
    colours = []
    for row in csv.reader(csvfile, delimiter=','):
         if len(row) <= 3:
              continue
         colours.append(row[3])

Answer 2

你可以用熊猫来实现这个代码。试试下面的代码：

import pandas as pd
df = pd.read_csv("input csv file")
df = df.T
df.to_csv("output csv file")

如果这是你要找的东西，更新一下

Answer 3

在不使用任何额外库的情况下，您可以编写接受特定列的函数，并将该列作为数组返回。

这样的函数看起来可能是这样的。

def column_to_row(col_num, csv):
    return list(row[col_num] for row in csv)

然后，您可以提取任何您想要的列，或者像这样迭代整个csv。

new_csv = []
for i in range(0, len(csv[0]):
    new_csv = new_csv.append(column_to_row(i, csv))