38 lines
1019 B
Python
38 lines
1019 B
Python
|
import csv
|
|||
|
|
from collections import namedtuple
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
# 想要读取csv数据，可以使用csv库
|
|||
|
|
with open("6.数据编码与处理/1.stocks.csv") as f:
|
|||
|
|
f_csv = csv.reader(f)
|
|||
|
|
headers = next(f_csv)
|
|||
|
|
print(headers)
|
|||
|
|
|
|||
|
|
for row in f_csv:
|
|||
|
|
print(row)
|
|||
|
|
|
|||
|
|
# 但是这样处理就非常不妙，所有的东西混在一起，变成奇奇怪怪的样子，不如使用named_tuple
|
|||
|
|
f.seek(0)
|
|||
|
|
|
|||
|
|
Row = namedtuple("Row", headers)
|
|||
|
|
for row in f_csv:
|
|||
|
|
print(row)
|
|||
|
|
row = Row(*row)
|
|||
|
|
print(row.Symbol)
|
|||
|
|
|
|||
|
|
f.seek(0)
|
|||
|
|
f_csv = csv.DictReader(f)
|
|||
|
|
for row in f_csv:
|
|||
|
|
print(row['Symbol'])
|
|||
|
|
|
|||
|
|
# 如果想要插入一条数据，可以使用writer
|
|||
|
|
rows = [
|
|||
|
|
("BB", 123, "9/12/2024", "9:36AM", -12, 999)
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
with open("6.数据编码与处理/1.stocks.csv", 'w+') as f:
|
|||
|
|
f_csv = csv.writer(f)
|
|||
|
|
f_csv.writerows(rows)
|
|||
|
|
|
|||
|
|
# 我的评价是不如pandas
|