从CSV导入的更有效方法

时间:2019-06-17 23:57:02

标签: python pandas

我想从CSV导入到对象。为简便起见,我们将其说成是一个城市,并且我有一个类似CSV的文件:

Seattle,WA,600,000,Seahawks,Starbucks

将其导入类的最佳方法是什么?现在,我导入CSV,并执行以下操作:

with open(filePath,'rb') as r:
    cityReader = csv.reader(r)
    for row in cityReader:
      cityName = row[0]
      state = row[1]
      population=row[2]
      nflTeam=row[3]
      bigCompany=row[4]
      newCity=city(cityName,state,population,nflTeam,bigCompany)
      addToCityList(newCity)

我想知道是否有更好的方法。我觉得您也许可以将熊猫用于这样的事情?这似乎并不是最有效的方法。

1 个答案:

答案 0 :(得分:-1)

尝试:-

import pandas as pd
df = pd.read_csv("/path/to/XXX.csv")
for idx, row in df:
    city_name=row[0]
    state = row[1]