时间戳

首页 > 解决方案 > pandas python中需要拆分的一列数据

python - pandas python中需要拆分的一列数据

问题描述

我有一个看起来像这样的数据框：

0 movie*
1 star wars
2 the godfather
3 zombieland
4 book*
5 romeo and juliet
6 harry potter
7 tv series*
8 breaking bad
9 game of thrones
...

所以类别后跟该类别中的项目，然后是其他类别，都在一个列中。假设这样的数据会持续一段时间，有许多不同的类别。

然后我想要将数据框定为：

  type        name

0 movie*      star wars   
1 movie*      godfather
2 movie*      zombieland
3 book*       romeo and juliet
4 book*       harry potter
5 tv series*  breaking bad
6 tv series*  game of thrones
...

我已经用 .endswith('*') 实现了一个布尔掩码，但不知道如何将它合并到一个新的数据框中。

标签： pythonpandas

解决方案

利用：

print (df)
               name
0            movie*
1         star wars
2     the godfather
3        zombieland
4             book*
5  romeo and juliet
6      harry potter
7        tv series*
8      breaking bad
9   game of thrones

首先为第一个位置创建新列，如果不匹配，DataFrame.insert则为缺失值，然后向前填充非缺失值，然后过滤掉两列中的相同值，最后删除：Series.whereSeries.str.endswithboolean indexingSeries.ne*Series.str.strip

df.insert(0, 'type', df['name'].where(df['name'].str.endswith('*')).ffill())
df = df[df['type'].ne(df['name'])].copy()
df['type'] = df['type'].str.strip('*')
print (df)
        type              name
1      movie         star wars
2      movie     the godfather
3      movie        zombieland
5       book  romeo and juliet
6       book      harry potter
8  tv series      breaking bad
9  tv series   game of thrones

推荐阅读