读取csv字符串的函数

Question

此问题与现在已删除的 earlier question 重复，现在已根据原始中提供的 cmets 进行编辑。

我正在尝试创建一个以 CSV 格式输入字符串的函数。例如"id,name,age,score\n1,Jack,NULL,12\n17,Betty,28,11"。

它应该返回下表：

id	name	age	score
1	Jack	NULL	12
17	Betty	28	11

它还应该删除有缺陷的行。有缺陷的行是当它的值 NULL 全部大写时 - 任何其他字符，如（0 到 9 或 a 到 z 或 A 到 Z）都是可以接受的。

上述输入字符串的最终输出应该是：

id	name	age	score
17	Betty	28	11

这是我使用 pandas 和 csv 包的代码。我需要在不使用任何这些包的情况下创建它。

def test(S):
    result = pd.DataFrame(csv.reader(S.splitlines()))
    new_header = result.iloc[0]
    result = result[1:]
    result.columns = new_header 
    df = result.select_dtypes(object)
    new_result = ~df.apply(lambda series: series.str.contains('NULL')).any(axis=1)
    f_result = result[new_result]
    return f_result

Answer 1

这里是如何做你说你想做的事。您对所需“表格”输出的描述有些模糊，所以我做出了最好的猜测。

def get_rows(data):
    rows = []
    for line in data.splitlines():
        fields = line.split(',')
        if not any(field == 'NULL' for field in fields):  # Not defective row.
            rows.append(fields)
    return rows


csv_string = 'id,name,age,score\n1,Jack,NULL,12\n17,Betty,28,11'
rows = get_rows(csv_string)

# Find longest item in each column.
widths = [max(len(item) for item in col) for col in zip(*rows)]

# Create a row of separators and make it the second row of the list.
separator_row = [width*'-' for width in widths]
rows.insert(1, separator_row)  # Insert following header row.

# Create a format specification for rows of table.
field_specs = [f' {{:{width}}} ' for width in widths]
format_spec = '|' + '|'.join(field_specs) + '|'

# Print formatted data.
for row in rows:
    print(format_spec.format(*row))

纯文本示例输出：

| id | name  | age | score |
| -- | ----- | --- | ----- |
| 17 | Betty | 28  | 11    |