处理数据中不必要的部分,并将处理过的数据写入新的文件中

import os 
import csv
import re
file_path =r'C:\Users\DKTE-assist2009.csv'
file_name =r'C:\Users\remove-DKTE-assist2009.csv'
if os.path.exists(file_path):
    print("文件存在")

with open(file_path, 'r', newline='') as file:
    reader = csv.reader(file)
    
    # 一次性读取整个文件内容并存储在一个列表中
    data = list(reader)
pattern = r'tensor\(([\d.]+),'

for i in range(1,len(data)):
    for j in range(21):
        match = re.search(pattern, data[i][j])
        data[i][j]=float(match.group(1))
with open(file_name, 'a', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)

r'C:\Users\DKTE-assist2009.csv'

"r" 表示这是一个原始字符串,其中包含了文件路径。在处理文件路径等可能包含反斜杠符号的字符串时,使用原始字符串是一种良好的做法,可以避免潜在的问题。

如果是逐行写入文件的话:就是writerows和writerow的区别

import os 
import csv
import re
file_path =r'C:\Users\DKTE-assist2009.csv'
file_name =r'C:\Users\remove-DKTE-assist2009.csv'
if os.path.exists(file_path):
    print("文件存在")

with open(file_path, 'r', newline='') as file:
    reader = csv.reader(file)
    
    # 一次性读取整个文件内容并存储在一个列表中
    data = list(reader)
with open(file_name, 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(data[0])

pattern = r'tensor\(([\d.]+),'

for i in range(1,len(data)):
    for j in range(21):
        match = re.search(pattern, data[i][j])
        data[i][j]=float(match.group(1))
    with open(file_name, 'a', newline='') as file:
        writer = csv.writer(file)
        writer.writerow(data[i])

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

铁灵

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值