CSV读取文件困难

Question

编写一个名为“ csv_to_kvs”的函数，该函数将字符串作为参数来表示具有4列的CSV文件名，格式为“ string，float，float，float”，并返回一个新的键-值存储，将字符串映射到浮点数。 返回的键值存储将为文件中的每一行提供一对，其中CSV文件的第一列中的键来自第三列中的值。 （下面的代码）

import csv
def csv_to_kvs(string):
    with open(string) as f:
        file = csv.DictReader(f)
        for column in file:
            for key in column.keys():
                return key
             for value in column.values():
                return value

当我提交以下函数csv_to_kvs ，输入错误。

输入experienced.csv ：

spite,-11.09,3.92,7.38
questionnaire,12.8,-4.39,-0.14
literally,19.5,-3.94,-5.06
colleague,17.19,-15.3,0.12

返回： "spite"

预期： {'colleague': -15.3, 'spite': 3.92, 'questionnaire': -4.39, 'literally': -3.94}

Answer 1

csv.DictReader希望标题是第一行的列，而不是第一列的行。

相反，您应该使用csv.reader读取行，使用zip进行转置，以将第一个项目作为标头解包，将其余的作为数据解压缩，然后使用列表推导来构造字典列表：

with open(string) as f:
    headers, *data = zip(*csv.reader(f))
    l = [{header: item for header, item in zip(headers, lst)} for lst in data]

用您的样本输入， l将变为：

[{'spite': '-11.09', 'questionnaire': '12.8', 'literally': '19.5', 'colleague': '17.19'}, {'spite': '3.92', 'questionnaire': '-4.39', 'literally': '-3.94', 'colleague': '-15.3'}, {'spite': '7.38', 'questionnaire': '-0.14', 'literally': '-5.06', 'colleague': '0.12'}]

第三列就是l[1] ：

{'spite': '3.92', 'questionnaire': '-4.39', 'literally': '-3.94', 'colleague': '-15.3'}

Answer 2

您可以使用csv.reader() 。 这可以帮助您将row作为list阅读。

import csv
def csv_to_kvs(string):
  with open(string) as f:
    my_file = csv.reader(f)
    my_dict = {}
    for row in my_file:
     # as you want to key value as the first one and value will be third one
       my_dict[row[0]] = row[2]
    return my_dict

输出： {'spite': '3.92', 'questionnaire': '-4.39', 'literally': '-3.94', 'colleague': '-15.3'}

Answer 3

Python Pandas is a great library to read and write CSV files

* Read CSV file - 

import pandas as pd
data = pd.read_csv("filename.csv")
data.head()

* Write CSV file - 

import pandas as pd
result = {Any key value pair data}
df = pd.DataFrame(data=result)
df.to_csv(filepath.csv, sep='\t', encoding='utf-8')