获取csv的一列并写入文件

以下是获取csv的一列并写入文件的Python代码示例：

import csv

# 打开csv文件
with open('data.csv', 'r') as file:
    # 读取csv文件内容
    reader = csv.reader(file)
    # 获取第二列数据
    column = [row[1] for row in reader]

# 写入文件
with open('column.txt', 'w') as file:
    # 将列数据写入文件
    for item in column:
        file.write("%s\n" % item)

在这个示例中，首先打开csv文件并使用csv模块的reader函数读取文件内容。然后，使用列表推导式获取第二列数据，并将其存储在变量column中。最后，打开一个新的文本文件并将列数据写入该文件。

相关问题

怎么在csv文件中定义新的一列并写入数据呢

将新的一列作为新增的一列，在csv文件中定义并写入数据，需要使用CSV库。下面是一个Python示例代码，可以向已有的csv文件中添加一列并写入数据：

import csv

# 读取csv文件
with open('example.csv', mode='r') as file:
    reader = csv.reader(file)
    rows = [row for row in reader]

# 如果csv文件已经有一个“新列”了，则不需要在次添加
if rows[0][-1] != '新列':
    rows[0].append('新列') #在表格最后添加新列的column name

# 在每一行添加数据
for row in rows[1:]:
    data = row[0] # 获取当前行的数据
    # 在新列中添加新的数据
    # 这里根据具体情况更新你的写入方式
    row[-1] = "new data"

# 将新文件写回csv
with open('example.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(rows)

以上代码首先读取csv文件，然后检查是否已经有了“新列”。如果没有，则在表头中添加“新列”。接着，在每一行中添加数据并将数据写回原文件。您可以根据自己的需要自定义代码来写入数据。

python按照某一列拆分csv文件

你可以使用Python的csv模块来读取和处理csv文件，然后根据某一列的值来拆分文件。以下是一个示例代码，将根据第二列的值来拆分文件：

import csv

# 读取csv文件
with open('input.csv', 'r') as csvfile:
    reader = csv.reader(csvfile)
    # 读取表头
    header = next(reader)
    # 创建一个字典，用于存储拆分后的文件
    split_files = {}
    for row in reader:
        # 获取第二列的值
        col_value = row[1]
        # 如果拆分文件不存在，则创建一个新的文件，并写入表头
        if col_value not in split_files:
            split_files[col_value] = open(col_value + '.csv', 'w', newline='')
            writer = csv.writer(split_files[col_value])
            writer.writerow(header)
        # 将当前行写入对应的拆分文件
        writer.writerow(row)
    # 关闭所有拆分文件
    for f in split_files.values():
        f.close()

在上面的代码中，我们首先使用csv.reader读取csv文件，并使用next函数跳过表头。然后，我们遍历每一行数据，并获取第二列的值。如果拆分文件不存在，则创建一个新的文件，并将表头写入文件。接着，我们将当前行写入对应的拆分文件。最后，我们使用close函数关闭所有拆分文件。

你可以根据需要修改代码，例如根据其他列来拆分文件，或者使用csv.DictReader来读取csv文件。