怎么将txt中的数据按41划分并生成两个文件
可以使用Python语言来实现这个功能。
假设我们有一个名为data.txt的文本文件,其中包含了一些数据,我们需要将这些数据按照4:1的比例进行划分,并将划分后的数据分别写入两个不同的文件。
以下是实现代码:
import random
# 读取数据
with open('data.txt', 'r') as f:
data = f.readlines()
# 随机打乱数据顺序
random.shuffle(data)
# 计算划分比例
split_ratio = 0.8 # 4:1的比例为0.8:0.2
# 计算划分点
split_point = int(len(data) * split_ratio)
# 将数据划分为两个部分
train_data = data[:split_point]
test_data = data[split_point:]
# 将划分后的数据写入文件
with open('train.txt', 'w') as f:
f.writelines(train_data)
with open('test.txt', 'w') as f:
f.writelines(test_data)
首先,我们使用with open语句读取data.txt文件中的数据,并将其保存到一个列表中。然后,我们使用random.shuffle函数将数据随机打乱顺序,以确保划分后的数据是随机的。
接着,我们计算出划分比例split_ratio,并根据该比例计算出划分点split_point。这里我们使用了Python内置的int函数来将计算得到的浮点数转换为整数,以确保split_point是一个整数。
最后,我们将划分后的数据分别写入train.txt和test.txt文件中,使用了with open语句打开文件,并使用f.writelines函数将数据写入文件。
运行该程序后,会生成两个文件train.txt和test.txt,分别包含了按照4:1比例划分后的训练数据和测试数据。
原文地址: https://www.cveoy.top/t/topic/bkg5 著作权归作者所有。请勿转载和采集!