[英]How can I delete all the columns in a csv file other than the ones I input?
all_filenames = [i for i in glob.glob("/home/greenvulcano/Scrivania/File_csv/*.csv")] dest_file = "/home/greenvulcano/Scrivania/Unione_files.csv" tag = ["inclination","deformation","temperature"] content = [] for file_name in sorted(all_filenames): with open(file_name, "r") as f: lines = f.readlines() for i, l in enumerate(lines): if len(content) == 0: content.append(l) elif i != 0: content.append(l) with open(dest_file, "w") as f: f.writelines(content) seleziona = input("Inserisci le colonne che vuoi visualizzare: ") with open(dest_file, "r") as f: titoli = f.readline().split(",") indici = [] for titolo in titoli: titolo_split = titolo.split('*') if seleziona in titolo_split: indice = titoli.index(titolo) indici.append(indice) righe = f.readlines() lista = [] for riga in righe: rigasplit = riga.split(",") for i in indici: lista.append(rigasplit[i])
I guess you can use:我想你可以使用:
test.csv测试文件
id,name,dob
1,pedro,2000-01-01
2,nuno,1999-01-01
import csv
csv_file = "test.csv"
seleziona = input("Inserisci le colonne che vuoi visualizzare: ").strip()
with open(csv_file) as f:
reader = csv.DictReader(f, delimiter=',')
for row in reader:
print(row[seleziona])
pedro
nuno
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.