如何從 pandas dataframe 列內的列表中提取特定數字模式

Question

我有如下數據框，我需要對其進行轉換

1. 從comma,前的第一個數字中提取並僅保留 3 個小數點，第二個數字也僅保留 3 個小數點。

2. 逗號應替換為:

3. 如果數字只有兩個小數點，再加一個零，使其成為小數點后三個。

輸入

  df
    [151.20732,-33.86785]
    [81.67732,-09.86]
    [1.2890,43.8] 
    [567.200,33.867]
    [557.21,33.86]

預計 Output

151.207:-33.867
81.677:-09.860
1.289:43.800
567.200:33.867
557.210,33.860

如何在 pandas 中做到這一點？

Answer 1

這比我想象的要難

def func(y,n):
    if y < 0 :
      return "%0.3f" % (-(y * 10 ** n // -1 / 10 ** n))
    else :
      return "%0.3f" % (y * 10 ** n // 1 / 10 ** n)
    

df.apply(lambda x : ':'.join([ func (y, 3) for y in x]) )
Out[86]: 
0    151.207:33.867
1      81.677:9.860
2      1.289:43.800
3    567.200:33.867
4    557.210:33.860
dtype: object

輸入

data = [[151.20732,-33.86785],
        [81.67732,-09.86],
        [1.2890,43.8],
        [567.200,33.867],
        [557.21,33.86]]
df = pd.Series(data)

Answer 2

DataFrame 輸入選項 1：

data = [[[151.20732,-33.86785]],
        [[81.67732,-09.86]],
        [[1.2890,43.8]],
        [[567.200,33.867]],
        [[557.21,33.86]]]
df = pd.DataFrame(data, columns=['geo'])

DataFrame 輸入選項 2：
literal_eval 用於讀取包含列表的 CSV 文件，否則將列表作為單個字符串讀取。

import ast

literal = lambda x: ast.literal_eval(x)
data = pd.read_csv('/Test_data.csv', converters={'geo.geometry.coordinates': literal})

df = pd.DataFrame(data, columns=['geo.geometry.coordinates'])
df.rename(columns = {'geo.geometry.coordinates':'geo'}, inplace = True)

算法：

import math

def trunc(f,d):
    # Truncate float (f) to d decimal places, unless NaN
    return 'nan' if math.isnan(f) else f"{int(f*10**d)/10**d:0.{d}f}"

df['geo_neo'] = df.apply(lambda r:  trunc(r['geo'][0], 3) + ':'
                                  + trunc(r['geo'][1], 3), axis = 1)

DataFrame Output：

                      geo          geo_neo
0  [151.20732, -33.86785]  151.207:-33.867
1       [81.67732, -9.86]    81.677:-9.860
2           [1.289, 43.8]     1.289:43.800
3         [567.2, 33.867]   567.200:33.867
4         [557.21, 33.86]   557.210:33.860