如何在数据框中添加列？

Question

我有以下数据：

例子：

DRIVER_ID;TIMESTAMP;POSITION

156;2014-02-01 00:00:00.739166+01;POINT(41.8836718276551 12.4877775603346)

我想创建一个带有 4 列的 pandas 数据框，它们分别是 id、时间、经度、纬度。 到目前为止，我得到了：

cur_cab = pd.DataFrame.from_csv(
            path,
            sep=";",
            header=None,
            parse_dates=[1]).reset_index()
cur_cab.columns = ['cab_id', 'datetime', 'point']

path 指定包含数据的 .txt 文件。 我已经编写了一个函数，它从点格式的字符串中返回经度和纬度值。 如何使用附加列和拆分值扩展数据框？

Answer 1

加载后，如果您使用的是最新版本的 pandas，则可以使用矢量化的 str 方法来解析列：

In [87]:
df['pos_x'], df['pos_y']= df['point'].str[6:-1].str.split(expand=True)
df

Out[87]:
   cab_id                   datetime  \
0     156 2014-01-31 23:00:00.739166   

                                      point  pos_x  pos_y  
0  POINT(41.8836718276551 12.4877775603346)      0      1  

另外你应该停止使用from_csv，它不再更新，使用顶级read_csv，这样你的加载代码将是：

cur_cab = pd.read_csv(
            path,
            sep=";",
            header=None,
            parse_dates=[1],
            names=['cab_id', 'datetime', 'point'],
            skiprows=1)