删除txt文件某列元素（python）

最新推荐文章于 2024-01-06 14:36:01 发布

访风景于崇阿

最新推荐文章于 2024-01-06 14:36:01 发布

阅读量3.8k

点赞数

CC 4.0 BY-SA版权

文章标签： python

本文链接：https://blog.csdn.net/starlight1231/article/details/123185229

该博客介绍了一个Python脚本，用于读取txt文件并删除最后一列，然后将处理后的数据写入新的txt文件。代码通过numpy和内置的文件操作函数实现，适用于数据预处理场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在数据使用前，需要进行数据预处理。

现在有一个txt文件内存有数据，希望删除其最后一列。

删除前删除后

实现代码：

import numpy as np
import os


def dataprocess():
    # read
    data = []
    with open('../3_cluster/test.txt', 'r') as f:
        for line in f.readlines():
            data.append(line[:-1].split(','))
    data = np.array(data)
    print(data.shape)

    # data process
    data2 = []
    for index, row in enumerate(data.T):
        if index < len(data.T)-1:
            data2.append(row)
    data2 = np.array(data2)
    print(data2)                    # 拿到了前2列数据

    # write
    a = data2.T
    with open('../3_cluster/test2.txt', 'w') as f:
        for i in range(len(a)):
            for j in range(0, 2):
                if j == 0:
                    f.write(str(a[i][j]) + ',')
                else:
                    f.write(str(a[i][j]) + '\n')


    # os.remove('../3_cluster/test.txt')
    # os.rename('../3_cluster/test2.txt', '../3_cluster/test.txt')


if __name__ == '__main__':
    dataprocess()

注：