为数据框列分配值

debugcn 发表于 Dev

用户名

在下面的代码中，未填充数据帧df5。我只是将值分配给dataframe的列，并且我已经预先指定了该列。当我打印数据框时，它返回一个空的数据框。不知道我是否想念一些东西。

任何帮助，将不胜感激。

import math    
import pandas as pd

columns = ['ClosestLat','ClosestLong']

df5 = pd.DataFrame(columns=columns)

def distance(pt1, pt2):
  return math.sqrt((pt1[0] - pt2[0])**2 + (pt1[1] - pt2[1])**2)

for pt1 in df1:
   closestPoints = [pt1, df2[0]]
   for pt2 in df2:
     if distance(pt1, pt2) < distance(closestPoints[0], closestPoints[1]):
       closestPoints = [pt1, pt2]
       df5['ClosestLat'] = closestPoints[1][0]
   df5['ClosestLat'] = closestPoints[1][0]
   df5['ClosestLong'] = closestPoints[1][1]
   print ("Point: " + str(closestPoints[0]) + " is closest to " + str(closestPoints[1]))

bpachev

从代码的外观来看，您试图填充df5纬度和经度的列表。但是，您犯了一些错误。

大熊猫数据框的列是Series，并保存某种类型的顺序数据。因此，df5['ClosestLat'] = closestPoints[1][0]尝试为整个列分配单个数值，并导致一个空列。
即使数据框没有忽略您为该列分配实数的尝试，您也会丢失数据，因为每次循环都会覆盖该列。

解决方案：建立经度和纬度的列表，然后插入数据框。

import math    
import pandas as pd

columns = ['ClosestLat','ClosestLong']

df5 = pd.DataFrame(columns=columns)

def distance(pt1, pt2):
  return math.sqrt((pt1[0] - pt2[0])**2 + (pt1[1] - pt2[1])**2)

lats, lngs = [], []
for pt1 in df1:
   closestPoints = [pt1, df2[0]]
   for pt2 in df2:
     if distance(pt1, pt2) < distance(closestPoints[0], closestPoints[1]):
       closestPoints = [pt1, pt2]
   lats.append(closestPoints[1][0])
   lngs.append(closestPoints[1][1])

df['ClosestLat'] = pd.Series(lats)
df['ClosestLong'] = pd.Series(lngs)

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。