python 2.7脚本添加一列

Question

我有一个这种格式的输入文件，有 5 列

输入文件：1.csv

INK137377383,3,2613,1513,899
INK133344444,3,106029,106961,12981 
INK112233343,2,1775,NA,NA 
INK22233333,3,1252,767,758
INK1223824,1,NA,NA,NA

我正在考虑的条件是

如果输入文件中第 2 列的值为 3 并且如果 col4,5 和 6 值小于或等于 3000 那么它应该追加 3_category到输出

如果输入文件中第 2 列的值为 3 并且如果 col4,5 和 6 值在 10000 和 15000 之间，那么它应该附加 3_1category 到输出

如果输入文件中第 2 列的值为 2 并且如果 col4,5 和 6 值小于或等于 3000 那么它应该追加 2_category 到输出

如果输入文件中第 2 列的值为 2 并且如果 col4,5 和 6 值在 10000 和 15000 之间，那么它应该附加 2_1category 到输出

如果输入文件中第 2 列的值为 1，那么它应该追加 1_category 到输出

预期输出：

INK137377383,3,2613,1513,899,3_category
INK133344444,3,106029,106961,12981,3_1category   
INK112233343,2,1775,NA,NA,2_category
INK22233333,3,1252,767,758,3_category
INK1223824,1,NA,NA,NA,1_category

这是我在 python 2.7 上尝试过但无法获得预期输出的脚本

import csv


with open('1.csv') as k1, open('out.csv', 'w') as k2:
    reader = csv.reader(k1)
    writer = csv.writer(k2)

    headers = next(reader)
    headers.append('new_column')
    writer.writerow(headers)

    for row in reader:
        new_value = compute_from_row(row)
        row.append(new_value)
        writer.writerow(row)

Answer 1

鉴于您提出的问题没有足够的信息，我已经制定了一个实施您所写规则的答案。不幸的是，这很难验证，因为您提供的预期输出不符合规则（例如 133344444,3,106029,106961,12981,3_1category）。我还假设如果找到 NA 值，则应将其视为零整数值。

def compute_from_row(row):
    # Replace NA with value of 0
    row = ['0' if 'NA' in x else x for x in row]
    # Convert to integers ignoring the first column
    row[1:] = map(int, row[1:])
    numbers = row[2:5]
    # Implement the rules
    if row[1] == 3 and all(map(lambda x: x<=3000, numbers)):
      return '3_category'
    elif row[1] == 3 and all(map(lambda x: x<15000 and x>10000, numbers)):
      return '3_1category'
    elif row[1] == 2 and all(map(lambda x: x<=3000, numbers)):
      return '2_category'
    elif row[1] == 2 and all(map(lambda x: x<15000 and x>10000, numbers)):
      return '2_1category'
    elif row[1] == 1:
      return '1_category'



import csv


with open('1.csv') as k1, open('out.csv', 'w') as k2:
    reader = csv.reader(k1)
    writer = csv.writer(k2)

    headers = next(reader)
    headers.append('new_column')
    writer.writerow(headers)

    for row in reader:
        new_value = compute_from_row(row)
        row.append(new_value)
        writer.writerow(row)