检查当前目录下是否有一个名为aa.csv的文件,如果没有,创建一个; 向aa.csv文件写入数据:第一列的第一行为spb_id,第一列的第二行的值为1,第一列的第二行的值为2,依次类推,知道400;第二例第一行为EC,第三列第一行为RC 创建两个字典:第一个字典的键分别为1到400,每个键对应的值为键+1,比如键为1,值就为2; 第二个字典的键分别为1到400,每个键对应的值为键+2,比如键为1,值就为3 把第一个字典的值依次写到aa.csv列名为EC的列,比如 键为1,把值填到spb_id列为1的EC列 把第2个字典的值依次写到aa.csv列名为RC的列,比如 键为1,把值填到spb_id列为1的RC列 用python实现
时间: 2024-02-18 13:02:05 浏览: 151
AA.rar_txt文件处理_文件数据处理_读入TXT数据
好的,以下是Python代码实现:
```python
import os
import csv
# 检查当前目录下是否有 aa.csv 文件,如果没有,则创建一个
if not os.path.exists('aa.csv'):
with open('aa.csv', 'w', newline='') as f:
writer = csv.writer(f)
writer.writerow(['spb_id', 'EC', 'RC'])
for i in range(1, 401):
writer.writerow([i, '', ''])
# 向 aa.csv 文件写入数据
with open('aa.csv', 'r+', newline='') as f:
reader = csv.reader(f)
header = next(reader)
for i, row in enumerate(reader, 1):
row[1] = i
row[2] = i + 1
f.seek(0)
writer = csv.writer(f)
writer.writerow(header)
writer.writerows(reader)
f.truncate()
# 创建两个字典
dict1 = {i: i+1 for i in range(1, 401)}
dict2 = {i: i+2 for i in range(1, 401)}
# 把第一个字典的值依次写到 aa.csv 列名为 EC 的列
with open('aa.csv', 'r+', newline='') as f:
reader = csv.reader(f)
header = next(reader)
for i, row in enumerate(reader, 1):
row[header.index('EC')] = dict1[i]
f.seek(0)
writer = csv.writer(f)
writer.writerow(header)
writer.writerows(reader)
f.truncate()
# 把第二个字典的值依次写到 aa.csv 列名为 RC 的列
with open('aa.csv', 'r+', newline='') as f:
reader = csv.reader(f)
header = next(reader)
for i, row in enumerate(reader, 1):
row[header.index('RC')] = dict2[i]
f.seek(0)
writer = csv.writer(f)
writer.writerow(header)
writer.writerows(reader)
f.truncate()
```
此代码可以实现检查当前目录下是否有 aa.csv 文件,如果没有则创建,然后向文件中写入数据并修改文件中 EC、RC 两列的数据。
阅读全文