将列添加到pandas dataframe&迭代其中一列

时间:2017-06-27 14:50:43

标签: python google-maps pandas gis geocode

我加载了一个包含多个列的数据框,其中一列包含一个地址。我正在使用python地理编码器模块来获取此csv中每个地址的纬度/经度。

熊猫

1)如何添加新列?我应该通过行来添加列,还是应该在开头添加列?

2)在下面的代码中,我试图遍历数据框中的每一行。对于每一行,我都在执行geocoder.google()方法。我的csv / data frame的第16列包含一个地址。

在迭代所有行时,我如何引用该地址列?如果我按原样运行代码,我会得到“IndexError:tuple index超出范围”。

CSV

3)我的代码的第二部分与CSV模块类似。我用CSV读取,遍历每一行并执行前面所述的地理编码方法。地理编码方法返回2个值的列表(2个坐标 - [XXXX,XXXX])。我试图写入原始行,然后再写两个列,然后是两个坐标。我得到“TypeError:只能连接列表(不是”浮动“)列表”

import geocoder 
import csv
import pandas as pd
import time

df = pd.read_csv("RSM100_1995.csv",header=None)
print(df.head())
for row in df.iterrows():
   g = geocoder.google(row[16])
   print(row[16],g.latlng)
   time.sleep(2)

with open("RSM100_1995.csv","r") as f, open("RSM_GCTest.csv","w",newline='') as g:
    rdr = csv.reader(f)
    wtr = csv.writer(g)
    for r in rdr:
        gc = geocoder.google(str(r[16]))
        print(r[16],gc.latlng)
        wtr.writerow(r + gc.latlng[0]+gc.latlng[1])
        time.sleep(2)

顺便说一句,我使用的是time.sleep(2),因为地理编码器对请求数量有限制。我没有像这里那样运行代码,只是把它放在这里显示它。

如果有人有更好的使用Python对英国地址进行地理编码的方法,请告诉我。

编辑:

对于Chirag - 我已经做了你提到的改变。我尝试使用相同的结果替换下面代码中的'Address'和列索引(16)。

我添加了带X.columns的列标题

我现在收到一条链接许多不同文件的非常长的错误消息。

RS1995 = pd.read_csv("RSM100_1995.csv",header=None)

RS1995.columns = ['ID','Price','Date','Postcode','X','Y','Z','PAON','SAON','Street','Locality','District','City','County','A','B','Address','XX']
print(RS1995.head())
for row in RS1995.iterrows():
    RS1995['lat'] = geocoder.google(RS1995['Address']).latlng[0]
    RS1995['lng'] = geocoder.google(RS1995['Address']).latlng[1]
    print(RS1995.head())
    time.sleep(2)

就CSV而言 - 有17列,我已在上面标题。 “地址”列是我想要通过地理编码器的列。地址栏本身是'PAON','SAON','Street','Locality','County'和&的串联。 “邮编”。我也可以包括'City',但我使用CSV模块进行的所有连接。

如果有帮助 - 这是Geocoder链接:

http://geocoder.readthedocs.io/

编辑2:

RS1995 = pd.read_csv("RSM100_1995.csv",header=None)

RS1995.columns = ['ID','Price','Date','Postcode','X','Y','Z','PAON','SAON','Street','Locality','District','City','County','A','B','Address','XX']
print(RS1995.head())

RS1995['lat'] = "x"
RS1995['lng'] = "y"
print(RS1995.head())
for row in RS1995.iterrows():
    print(row)

每当我运行上面的代码时,我都会这样做。我刚才把最后两个作为例子。这是什么意思?我将如何在每一行中进行整理,对地址进行地理编码并等待2秒,以便我不超过速率限制?:

(98, ID                     {40E4DAC0-863F-42FE-94B4-49A70D3BE0B9}
Price                                                   43000
Date                                         24/02/1995 00:00
Postcode                                             WS12 3XJ
X                                                           S
Y                                                           N
Z                                                           F
PAON                                                        1
SAON                                                      NaN
Street                                           WOODFORD WAY
Locality                                          HEATH HAYES
District                                              CANNOCK
City                                            CANNOCK CHASE
County                                          STAFFORDSHIRE
A                                                           A
B                                                           A
Address     1  WOODFORD WAY HEATH HAYES STAFFORDSHIRE WS12...
XX          1  WOODFORD WAY HEATH HAYES STAFFORDSHIRE WS12...
lat                                                         x
lng                                                         y
Name: 98, dtype: object)
(99, ID                  {061625F8-82D5-43CF-A55F-4288979D31EC}
Price                                                42995
Date                                      01/09/1995 00:00
Postcode                                           PO1 5AY
X                                                        T
Y                                                        N
Z                                                        F
PAON                                                    67
SAON                                                   NaN
Street                                        BYERLEY ROAD
Locality                                        PORTSMOUTH
District                                        PORTSMOUTH
City                                            PORTSMOUTH
County                                          PORTSMOUTH
A                                                        A
B                                                        A
Address     67  BYERLEY ROAD PORTSMOUTH PORTSMOUTH PO1 5AY
XX          67  BYERLEY ROAD PORTSMOUTH PORTSMOUTH PO1 5AY
lat                                                      x
lng                                                      y
Name: 99, dtype: object)

1 个答案:

答案 0 :(得分:0)

您可以在pandas数据框中创建新列,类似于使用关联数组或字典的方式。您可以为纬度和经度创建两个新列:

df['lat'] = geocoder.google(df[16]).latlng[0]
df['lng'] = geocoder.google(df[16]).latlng[1]

然后您可以将整个数据帧写入csv:

df.to_csv('RSM_GCTest.csv')