几周前因为需要写的一个CSV简单的操作脚本,在实现智能化的数据交互方面还是有一些作用的(嵌入其它脚本里面可以进行模块导入),在博客上存储一下,以防以后用到。对于CSV的编辑更改功能我给去掉了,可以自己写一下,我的思路就是把按照用户所需要更改的位置进行查找更改,下面是一些简单的功能,还希望请各位大牛指导(测试运行方法:可以直接拷贝进IDLE文件里面(不要交互式)运行即可)
#建立csv.py
import os
def init(name):
k=input("请输入您想要输入的行数(直接回车为退出):")
if k:
try:
num=int(k)
except ValueError as e:
print(e)
else:
text=open("{0}".format(name),"w+")
for i in range(num):
t=()
'''
t=[]
t.append(input("请输入类别:"))
t.append(input("请输入数字1:"))
t.append(input("请输入数字2:"))
t.append(input("请输入数字3:"))
t.append(input("请输入数字4:"))
t.append(input("请输入数字5:"))
'''
t=input("请输入你想要输入的数值(数据之间用逗号隔开):")
text.write("".join(t)+"\n")
text.close()
print("复写成功!")
def read(name):
text=open("{0}".format(name),"r+")
for i in text:
i=i.replace("\n","")
print(i)
text.close()
def add(name):
k=input("请输入您想要输入的行数(直接回车为退出):")
if k:
try:
num=int(k)
except ValueError as e:
print(e)
else:
text=open("{0}".format(name),"a+")
for i in range(num):
t=()
'''
t=[]
t.append(input("请输入类别:"))
t.append(input("请输入数字1:"))
t.append(input("请输入数字2:"))
t.append(input("请输入数字3:"))
t.append(input("请输入数字4:"))
t.append(input("请输入数字5:"))
'''
t=input("请输入你想要输入的数值(数据之间用逗号隔开):")
text.write("".join(t)+"\n")
text.close()
print("添加成功!")
def main():
chose=1
while chose:
try:
name=input("请输入你想要写入或追加或覆盖重写的文件的路径以及的文件名(格式为xx盘\\xx\\xx\\xxx.csv,注:默认文件路径为脚本所在位置):")
print("本脚本实现以下功能:")
print("1、对已创建的csv文件进行覆盖重写操作\n2、对创建的csv文件进行读操作\n3、对已创建的csv文件进行尾追加操作\n4、注:所有操作都会在无此文件的情况下自动创建!")
try:
chose=int(input("请输入你选择的序号(输入0退出):"))
except ValueError as e:
print(e)
continue
#os.system("clear")
if chose==1:
init(name)
if chose==2:
read(name)
if chose==3:
add(name)
except:
print("文件路径错误!")
continue
print("bey!")
if __name__=='__main__':
main()