📜  python导入多个csv - Python(1)

📅  最后修改于: 2023-12-03 15:04:39.136000             🧑  作者: Mango

Python导入多个csv

在数据分析和机器学习领域,我们通常需要处理大量的数据文件。很多情况下,我们需要从多个csv文件中导入数据并进行预处理和分析。在这篇文章中,我将介绍如何使用Python导入多个csv文件。

导入单个csv文件

我们先回顾一下如何导入单个csv文件。Python的pandas库提供了强大的功能,可以方便地导入和处理csv文件。我们可以使用pandas的read_csv函数导入单个csv文件。以下是一个例子:

import pandas as pd

df = pd.read_csv('data.csv')

以上代码中,我们使用pd.read_csv函数读取名为'data.csv'的csv文件,并将其存储为pandas的DataFrame对象。这样,我们便可以方便地对数据进行处理和分析。

导入多个csv文件

接下来,我们将介绍如何导入多个csv文件。如果我们要手动导入多个csv文件,我们需要一个一个地读取它们。但是,这样很繁琐,代码也不好维护。因此,我们需要一个更好的方法。

在Python中,我们可以使用os库和glob库来自动化多个csv文件的导入。以下是一个例子:

import os
import glob
import pandas as pd

os.chdir('/path/to/csv/files') # 设置csv文件所在的目录

extension = 'csv'
all_filenames = [i for i in glob.glob('*.{}'.format(extension))] # 获取所有csv文件的文件名

# 将所有csv文件合并为一个DataFrame对象
combined_csv = pd.concat([pd.read_csv(f) for f in all_filenames ])

# 将合并后的DataFrame对象保存为新的csv文件
combined_csv.to_csv("combined_csv.csv", index=False, encoding='utf-8-sig')

以上代码中,我们使用os.chdir函数指定有多个csv文件的目录。然后,我们使用glob库获取该目录中所有扩展名为csv的文件的文件名。接下来,我们使用pd.read_csv函数读取每个csv文件,并使用pd.concat函数将它们合并为一个DataFrame对象。最后,我们使用combined_csv.to_csv函数将合并后的DataFrame对象保存为一个名为'combined_csv.csv'的新csv文件。

总结

本文介绍了如何使用Python导入多个csv文件。我们可以使用pandas的read_csv函数导入单个csv文件,并使用os库和glob库自动化多个csv文件的导入。使用以上方法,我们可以轻松地处理多个csv文件中的数据。