python - CSV 到 Python 中的列表

Question

我已将 .csv 文件导入到 Python 中，并将其命名为 data。它看起来像这样:

aK    bK    id
.124  .121  1
.320  .202  2
.222  .198  3

以此类推数百行。如何将 aK 列和 bK 列转换为列表？我已经尝试过:

aK = []
for row in data:
    aK.append(row[0])

但是aK仍然是一个空白列表。

Answer 1

import csv

CSV_FILE = "test.csv"

def main():
    with open(CSV_FILE) as inf:
        in_csv = csv.reader(inf, delimiter=" ", skipinitialspace=True)
        header = next(in_csv)       # skip header row
        ak, bk, id = zip(*in_csv)   # get columns

if __name__ == "__main__":
    main()

这给出了

ak = ('.124', '.320', '.222')
bk = ('.121', '.202', '.198')
id = ('1', '2', '3')