Python读csv文件去掉一列后再写入新的文件实例

以下是详细讲解“Python读csv文件去掉一列后再写入新的文件实例”的完整攻略：

1. 导入必要的库

在处理csv文件时，需要使用Python内置的csv库来读取和写入文件，因此第一步就是导入csv库：

import csv

2. 读取csv文件

接下来，我们需要使用csv库中的csv.reader()方法来读取csv文件。例如，假设我们要读取名为input.csv的csv文件，代码示例如下：

with open('input.csv', newline='') as csvfile:
    reader = csv.reader(csvfile)
    rows = [row for row in reader]

代码中，csv.reader()方法返回一个可迭代的CSV读取器对象，我们可以使用它来遍历csv文件的每一行数据，然后将数据存储到一个列表rows中。需要注意的是，使用with open()打开csv文件时，需要将newline=''设置为避免读取到不必要的空行。

3. 去掉一列数据

读取csv文件之后，我们需要以某种方式处理数据，例如去掉某一列数据。在本例中，我们可以使用列表推导式和切片操作来去掉第三列数据。代码示例如下：

new_rows = [[row[0], row[1], row[3]] for row in rows]

代码中，new_rows表示去掉第三列数据后的新数据集合，每一行数据依然是一个列表，只不过第三列已经被剔除掉了。

4. 写入新的csv文件

最后一步，我们需要将新数据集合写入到一个新的csv文件中。代码示例如下：

with open('output.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    for row in new_rows:
        writer.writerow(row)

代码中，csv.writer()方法返回一个CSV写入器对象，我们可以使用它将新数据集合写入到一个名为output.csv的csv文件中。使用newline=''同样是为了避免写入到csv文件中出现额外的空行。在写入新的csv文件时，我们需要使用writer.writerow()方法将每一行数据写入到文件中。

示例

下面我们来通过两个示例来说明上述过程。

示例1：

假设我们有一个csv文件scores.csv，数据如下：

name,math,english,physics
Alice,90,80,85
Bob,85,90,95
Charlie,92,78,88

我们要去掉英语这一列数据，然后将结果写入到new_scores.csv文件中。代码如下：

import csv

with open('scores.csv', newline='') as csvfile:
    reader = csv.reader(csvfile)
    rows = [row for row in reader]

new_rows = [[row[0], row[1], row[3]] for row in rows]

with open('new_scores.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    for row in new_rows:
        writer.writerow(row)

示例2：

假设我们有一个csv文件data.csv，数据如下：

name,age,gender,city
John,25,Male,New York
Lily,23,Female,San Francisco
David,30,Male,Los Angeles

我们要去掉城市这一列数据，然后将结果写入到new_data.csv文件中。代码如下：

import csv

with open('data.csv', newline='') as csvfile:
    reader = csv.reader(csvfile)
    rows = [row for row in reader]

new_rows = [[row[0], row[1], row[2]] for row in rows]

with open('new_data.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    for row in new_rows:
        writer.writerow(row)

以上就是“Python读csv文件去掉一列后再写入新的文件实例”的完整攻略，希望能对您有所帮助。