pandas 把数据写入txt文件每行固定写入一定数量的值方法

Pandas 是一个流行的 Python 数据分析工具，在数据分析过程中，我们通常需要将分析结果保存成文件。Pandas 支持将数据保存到多种格式的文件中，包括 CSV、Excel、JSON、SQL、以及纯文本文件等。在本文中，我们将介绍如何使用 Pandas 将数据保存到纯文本文件，并控制每行写入的数据数量。

安装 Pandas

在开始之前，我们需要先安装 Pandas。在终端中输入以下命令可以安装最新版的 Pandas：

pip install pandas

将数据写入纯文本文件

使用 Pandas 将数据写入纯文本文件非常简单，只需要调用 Pandas 提供的 to_csv() 方法，并指定文件名和其他的一些选项即可。比如，将某个 DataFrame 写入到 data.txt 文件中：

import pandas as pd

data = {'name': ['Alice', 'Bob', 'Charles', 'David'],
        'age': [25, 30, 35, 40],
        'income': [3000, 4000, 5000, 6000]}
df = pd.DataFrame(data)

df.to_csv('data.txt', index=False, header=True, sep='\t')

在上述代码中，我们传给 to_csv() 方法的参数包括以下几个：

• data.txt：要写入的文件名。
• index=False：不在输出中包含行索引。
• header=True：输出包含表头。
• sep='\t'：使用制表符作为字段分隔符。

执行以上代码后，将会在当前目录下生成一个名为 data.txt 的文件。文件内容如下：

name    age     income
Alice   25      3000
Bob     30      4000
Charles 35      5000
David   40      6000

可以看到，Pandas 默认每行写入所有的数据。

控制每行写入的数据数量

有时候，我们需要将数据按照固定的数量分成若干组，每组写入一行。Pandas 提供了 chunksize 参数来控制每行写入的数据数量。比如，将每行写入两个数据，可以这样写：

import pandas as pd

data = list(range(10))
df = pd.DataFrame(data)

with open('data.txt', 'w') as f:
    for chunk in pd.read_csv(df, chunksize=2):
        line = '\t'.join(map(str, chunk.values.flatten())) + '\n'
        f.write(line)

在这个例子中，我们首先将数据保存到一个 DataFrame 对象中，然后通过 pd.read_csv() 方法读取 DataFrame 并设置 chunksize 参数为 2，这样读取出来的 DataFrame 每次就只有两行。我们循环遍历每个读取的 DataFrame，将其展平成一维数组，并用制表符连接每个值，最终拼成一行字符串，写入到文件中。

文件内容如下：

0   1
2   3
4   5
6   7
8   9

我们可以看到，每行写入的数据量都被控制在了两个以内。

综上，本文介绍了如何使用 Pandas 将数据写入到纯文本文件，并控制每行写入的数据数量。