Python：我无法从函数中的标准输入解析 csv

Posted 2023-03-25

【中文标题】Python：我无法从函数中的标准输入解析 csv

【英文标题】：Python: I can't parse a csv from stdin in function

【发布时间】：2020-08-25 12:58:00

【问题描述】：

无法从函数中的标准输入解析 csv

您好，我在我的 Mac 上尝试使用 Pythons 的 csv 模块来解析通过 stdin 传递的空格分隔文件：

printf "2020-01-01 Ben 2\n2020-02-01 Jenny 4\n" | ./tmp.py

当我使用 working 函数运行代码时，我得到了预期的结果：

$ printf "2020-01-01 Ben 2\n2020-02-01 Jenny 4\n" | ./tmp.py
function: working
['2020-01-01', 'Ben', '2']
['2020-02-01', 'Jenny', '4']

当我使用最后调用的not_working 函数运行它时，我得到一个错误：

$ printf "2020-01-01 Ben 2\n2020-02-01 Jenny 4\n" | ./tmp.py
function: not_working
Traceback (most recent call last):
  File "./tmp.py", line 36, in <module>
    not_working() # if working() was here, it would work
  File "./tmp.py", line 27, in not_working
    print_csv(args.infile, delimiter=' ')
  File "./tmp.py", line 20, in print_csv
    reader = csv.reader(infile, delimiter)
_csv.Error: unknown dialect

这是最后调用not_working 的最小示例

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import argparse
import csv
import sys

parser = argparse.ArgumentParser()
parser.add_argument(
    "infile",
    nargs="?",
    type=argparse.FileType("r"),
    default=sys.stdin,
)
args = parser.parse_args()

def print_csv(infile, delimiter):
    reader = csv.reader(infile, delimiter)
    for row in reader:
        print(row)

def not_working():
    print("function: not_working")
    with args.infile:
        print_csv(args.infile, delimiter=' ')

def working():
    print("function: working")
    with args.infile:
        reader = csv.reader(args.infile, delimiter=" ")
        for row in reader:
            print(row)

not_working() # if working() was here, it would work
# working()

为什么会这样？为什么我可以在函数外部解析 CSV，但不能在函数内部解析？

【参考方案1】：

您忘记将delimiter 作为print_csv 中的关键字参数传递；这意味着您实际上是在调用csv.reader(infile, dialect=delimiter)，而“”是一种未知方言（通常仅限于'excel'、'excel-tab' 或'unix'。

请参阅csv.reader 方法签名的文档以获取更多详细信息。