我需要编写一个函数,在给定CSV文件名作为字符串和列号的情况下,将创建一个在该文件的特定列中具有不同值的数组。假设我的第一列是0就像在Python列表中一样。该文件还有一个页眉和页脚行。
import csv
def facet(file, column):
""" (str, int) -> array
Gets a list of distinct values from the specified column in the input file.
"""
答案 0 :(得分:0)
尝试以下方法:
import csv
def facet(file, column):
with open(file, newline='') as f_input:
return [row[column] for row in csv.reader(f_input)][1:-1]
print(facet('input.csv', 1))
将文件读取为csv
并使用列表推导仅从每行中选择单元格并构建列表。最后[1:-1
用于跳过文件中的页眉和页脚。