0

如何将具有 3 条数据（在相邻列中）的名称的 CSV 文件转换为以相邻数据作为参数的已创建类的变量？

下面的代码似乎创建了对象，但列表是对象位置，而不是我喜欢的名称。

任何人都可以帮忙吗？

class Teams(object):
  def __init__(self, TeamName, FT, FG, Three):
    self.TeamName = TeamName
    self.FT = FT
    self.FG = FG
    self.Three = Three

North_Carolina = Teams("North Carolina", .643,.458,.371)

print North_Carolina.TeamName, North_Carolina.FT, North_Carolina.FG,North_Carolina.Three
## works fine but manually typed data

def getData(fname):  ##fname is a csv file with 4 columns and 348 rows. One header row.
    Data = open(fname, 'r')
    TeamList = []
    #print Data.read() ## prints a list with comma separated values and line return
    for line in Data:
      info = line.split(",") ##info is a list of lists of Data with Team,FT,FG,3P
      name = info[0]
      FT = info[1]
      FG = info[2]
      Three = info[3]
      newTeam = (name, FT, FG, Three)
      newTeam2 = Teams(newTeam[0],newTeam[1],newTeam[2],newTeam[3]) ##newTeam2 is an object. 
      TeamList.append(newTeam2)
    print TeamList[1].TeamName ##prints North Carolina list[0] print header
    print TeamList #prints list of object locations but not objects themselves

getData("Stats 01-04-2013.csv")

第一print条语句打印：

North Carolina

第二条print语句打印：

[<__main__.Teams object at 0x02953230>, <__main__.Teams object at 0x029532B0>, etc etc
4

4 回答 4

1

我会使用 csv 模块来做这样的事情，主要基于模块文档中的示例。可能需要调整，这只是我的想法，但在处理这些类型的文件时我总是使用相关的。

import csv

with open('teams.csv', 'rb') as csvfile:
    teamreader = csv.reader(csvfile)
    for row in teamreader:
        newTeam = Teams(row[0], row[1], row[2], row[3])
        teamList.append(newTeam)
于 2013-01-05T00:03:07.113 回答
0

对象位置是对象的打印方式。严格来说，调用str或repr对列表中的每个元素进行list调用。repr默认情况下，repr(team)对象Teams会返回类似<main.Teams object at 0x02953230>. 您可以通过在类上定义__repr__方法来覆盖它。Teams但是，这听起来不像您想要的。

如果TeamList是Teams对象列表，并且您想将其转换为TeamName成员列表，则只需使用 alist comprehension将其转换：

TeamNameList = [team.TeamName for team in TeamList]

请注意，这仍然不是您想要打印的内容：

>>> print TeamNameList
['North Carolina', 'South Carolina', 'West Carolina', 'East Carolina', 'Underground Carolina', 'Cloud-level Carolina', 'Past Carolina', 'Future Carolina'] # those are all states, right?

你可能想要这样的东西：

>>> print ', '.join(TeamNameList)
North Carolina, South Carolina, West Carolina, East Carolina, Underground Carolina, Cloud-level Carolina, Past Carolina, Future Carolina

从您的评论中：

后面还有一个函数 simGame(teamA, Teamb)

我看到了这些实例，但是我不确定如何在第二个函数中调用它们。我本来打算直呼他们的名字。在我给出的示例中，North_carolina 可以传递给后面的（未显示的函数），并且可以对数据运行计算

我想我明白你在这里想要什么。您希望能够模拟北卡罗来纳州和南卡罗来纳州之间的游戏，而您所拥有的只是TeamList.

为此，您可能需要创建一个dict，将名称映射到Teams对象，如下所示：

TeamDict = {team.TeamName: team for team in TeamList}

现在，您可以执行以下操作：

simGame(TeamDict['North Carolina'], TeamDict['South Carolina'])

现在，simGame将获取类的北卡罗来纳州和南卡罗来纳州实例Teams作为其teamA和teamB参数，因此它可以执行以下操作：

def simGame(teamA, teamB):
    scoreA = int(teamA.FT * 1 * 20) + int(teamA.FG * 2 * 40) + int(teamA.Three * 3 * 10)
    scoreB = int(teamB.FT * 1 * 20) + int(teamB.FG * 2 * 40) + int(teamB.Three * 3 * 10)
    if scoreA > scoreB:
        print 'Home team {} beats visitor {} by {}!'.format(teamA.TeamName,
                                                            teamB.TeamName,
                                                            scoreA - scoreB)
    else:
        print 'Visitors {} win by {} over {} at home!'.format(teamB.TeamName,
                                                              scoreB - scoreA,
                                                              teamA.TeamName)

那是你要的吗？

一些额外的评论：

您也可以使用 执行与列表推导相同的操作map，这样可以避免编写team两次，但这也意味着您不能使用正常的表达式语法：

TeamNameList = map(operator.attrgetter('TeamName'), TeamList)

或者你可以使用mapwithlambda来取回表达式语法……和重复team：

TeamNameList = map(lambda team: team.teamName, TeamList)

但是对于这样的情况，列表推导通常被认为是 Python 的方式。（此外，如果你使用 Python 3，它不会改变，而map从 a 更改list为迭代器，这意味着print TeamNameList会给你类似的东西<builtins.map at 0x108022350>......但', '.join(TeamNameList)仍然可以工作。）

作为旁注，在标准（PEP 8）Python 风格中，通常只有类是这样的 TitleCase，变量和属性是小写的。如果你真的很喜欢 CamelCase，你可以使用 lowerFirstCamelCase，但是使用 TitleCase 会让阅读你的代码的人望而却步，他们会立即尝试找出TeamName类的定义位置。另请参阅 bvukelic 的评论。

作为另一个方面说明，你似乎做了很多重复的代码，我不知道为什么：

  name = info[0]
  FT = info[1]
  FG = info[2]
  Three = info[3]
  newTeam = (name, FT, FG, Three)

你只是复制info到newTeam; 为什么要添加所有不再使用的中间变量？

  newTeam2 = Teams(newTeam[0],newTeam[1],newTeam[2],newTeam[3])

您可以将所有这些替换为：

  newTeam2 = Teams(info[0],info[1],newTeam[2],newTeam[3])

甚至：

  newTeam2 = Teams(*info)

如果您在某个地方需要未向我们展示的单独变量，那很好，但您仍然不可能需要newTeam; 这样做：

  newTeam2 = Teams(name, FT, FG, Three)
于 2013-01-04T23:28:30.057 回答
0

只是为了涵盖对该问题的另一种解释：

class Teams(object):
    def __init__(self, TeamName, FT, FG, Three):
        self.TeamName = TeamName
        self.FT = FT
        self.FG = FG
        self.Three = Three

    def __str__(self):
        return str(self.TeamName)

    def __repr__(self):
        return self.__unicode__()

每当您打印对象时，它将显示为TeamName

于 2013-01-04T23:46:59.750 回答
0

__str__()通过为它们提供自定义和/或__repr__()转换方法来控制类实例的显示方式。这是编写代码的一种方法，它也可以遵循PEP 8 Style Guide中的许多建议。正如其他人已经建议的那样，您还可以通过使用csv模块读取文件来进一步改进或简化它，但我将主要关注您问题中显示的代码。

class Team(object):
    def __init__(self, team_name, free_throws, field_goals, three_pointers):
        self.team_name = team_name
        self.free_throws = free_throws
        self.field_goals = field_goals
        self.three_pointers = three_pointers

    def __repr__(self):
        return '{}({!r}, {}, {}, {})'.format(self.__class__.__name__,
                                             self.team_name, self.free_throws,
                                             self.field_goals, self.three_pointers)

def get_data(fname):  # fname is a csv file with 4 columns and one header row
    teams = []
    with open(fname, 'rt') as csv_file:
        csv_file.next()  # skip header row
        for line in csv_file:
            info = line.strip().split(",")
            teams.append(Team(*info))  # create a Team instance and add it to list
    return teams

print get_data("Stats 01-04-2013.csv")

样本输出：

[Team('North Carolina',  .643,  .458,  .371), Team('Michigan',  .543,  .358,  .271)]
于 2013-01-05T03:08:26.060 回答