Python函数创建一个特定列的csv文件的数组

时间:2016-10-09 22:32:26

标签: python arrays python-3.x csv

我需要编写一个函数,在给定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.
"""

1 个答案:

答案 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用于跳过文件中的页眉和页脚。