python - 在 Python 中从 csv 文件中生成对象

Question

我有一个 csv 文件 paises.csv，如下所示:

Chile;17000000;0.72;FALSE;0
Hanslandia;2000000;0.2;FALSE;91
Bastiland;4000000;0;TRUE;0

第一列是国家名称，以下是它的一些特征，例如人口。因此，我创建了一个 country 类，并尝试创建国家/地区并使用 csv 文件中的功能填充它:

# -*- coding: utf-8 -*-
##Importing libraries
import csv
##Defining classes
class country:
    def __init__(self,inhabitants,tolerance,underground,infected_perc):
        self.inhabitants = inhabitants
        self.tolerance = tolerance
        self.underground = underground
        self.infected_perc = infected_perc

with open('paises.csv', 'r') as f:
    reader = csv.reader(f, delimiter = ";")
    for row in reader:
        row[0]=country(row[1],row[2],row[3],row[4]) 

代码运行。然而，它甚至没有创建国家。当我尝试查看行时，我得到的是这样的:

AttributeError("Can't get attribute 'country' on ",)

Answer 1

我认为您需要一个容器来存储您正在创建的国家/地区。我还会考虑将类名大写:

import csv

class Country:
    def __init__(self,inhabitants,tolerance,underground,infected_perc):
        self.inhabitants = inhabitants
        self.tolerance = tolerance
        self.underground = underground
        self.infected_perc = infected_perc

countries = {}
with open('paises.csv', 'r') as f:
   reader = csv.reader(f, delimiter = ";")
   for row in reader:
       countries[row[0]] = Country(row[1], row[2], row[3], row[4])

您想要的国家/地区存储在字典countries中，您可以使用其名称作为键来访问它们。例如，要访问智利，您可以使用countries['Chile']。

更新:

如果您想使用列表而不是字典，您应该向类中添加一个字段来存储国家/地区名称:

import csv

class Country:
    def __init__(self,name, inhabitants,tolerance,underground,infected_perc):
        self.name =  name
        self.inhabitants = inhabitants
        self.tolerance = tolerance
        self.underground = underground
        self.infected_perc = infected_perc

countries = []
with open('paises.csv', 'r') as f:
   reader = csv.reader(f, delimiter = ";")
   for row in reader:
       countries.append(Country(row[0], row[1], row[2], row[3], row[4]))

如果使用列表作为容器，您可以替换 countries.append(Country(row[0], row[1], row[2], row[3], row[4])) by countries.append(Country(*row))