如何在python ubuntu中从csv文件中打印两个列数据集

Question

我有包含13列的csv数据文件。我想拆分两个列值，另一个列要再添加一列。

def rowfilter(col1, col2):
    try:
        csv_read = csv.reader(open('items.csv'), delimiter=',')
        for row in csv_read:
        print(row[int(cols[0]):int(cols[1])])  
    except FileNotFoundError:
        print("File not found")

    inputrows = input("Enter columns in the format: col1 col2 ")

    rowfilter(inputrows.split())) 

Answer 1

要使用csvs，通常最简单的方法是使用pandas模块。它提供易于使用的I / O，选择和操作界面。

def rowfilter(col1, col2):
    try:
        df = pd.read_csv('items.csv')

        print(df[[col1, col2]])  
    except FileNotFoundError:
        print("File not found")

    inputrows = input("Enter columns in the format: col1 col2 ")

    rowfilter(inputrows.split())) 

这将打印列。