服务器之家

服务器之家 > 正文

Pandas实现一列数据分隔为两列

时间:2020-05-19 10:02     来源/作者:Dennis_Shaw

分割成一个包含两个元素列表的列

对于一个已知分隔符的简单分割(例如,用破折号分割或用空格分割).str.split() 方法就足够了 。 它在字符串的列(系列)上运行,并返回列表(系列)。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
>>> import pandas as pd
>>> df = pd.DataFrame({'AB': ['A1-B1', 'A2-B2']})
>>> df
 
  AB
0 A1-B1
1 A2-B2
>>> df['AB_split'] = df['AB'].str.split('-')
>>> df
 
  AB AB_split
0 A1-B1 [A1, B1]
1 A2-B2 [A2, B2]

分割成两列,每列包含列表的相应元素

下面来看下如何从:分割成一个包含两个元素列表的列至分割成两列,每列包含列表的相应元素。

?
1
2
3
4
5
6
7
8
9
10
11
>>> df['AB'].str[0]
 
0 A
1 A
Name: AB, dtype: object
 
>>> df['AB'].str[1]
 
0 1
1 2
Name: AB, dtype: object

因此可以得到

?
1
2
3
4
5
6
7
8
9
10
11
>>> df['AB'].str.split('-', 1).str[0]
 
0 A1
1 A2
Name: AB, dtype: object
 
>>> df['AB'].str.split('-', 1).str[1]
 
0 B1
1 B2
Name: AB, dtype: object

可以通过如下代码将pandas的一列分成两列:

?
1
2
3
4
5
6
>>> df['A'], df['B'] = df['AB'].str.split('-', 1).str
>>> df
 
  AB AB_split A B
0 A1-B1 [A1, B1] A1 B1
1 A2-B2 [A2, B2] A2 B2

补充知识:pandas某一列中每一行拆分成多行的方法

在处理数据过程中,常会遇到将一条数据拆分成多条,比如一个人的地址信息中,可能有多条地址,既有家庭地址也有工作地址,还有电话信息等等类似的情况,实际使用数据的时候又需要分开处理,这个时候就需要将这一条数据进行拆分成多条,以方便使用。

在pandas中如何对DataFrame进行相关操作呢,经查阅相关资料,发现了一个简单的办法,

info.drop(['city'], axis=1).join(info['city'].str.split(' ', expand=True).stack().reset_index(level=1, drop=True).rename('city'))

看起来非常之长,分开来看,流程如下:

将需要拆分的数据使用split拆分工具拆分,并使用expand功能拆分成多列

将拆分后的多列数据进行列转行操作(stack),合并成一列

将生成的复合索引重新进行reset保留原始的索引,并命名

将上面处理后的DataFrame和原始DataFrame进行join操作,默认使用的是索引进行连接

具体操作如下:

预操作:生成需要使用的DataFrame

?
1
2
3
4
5
6
# 用来生成DataFrame的工具
from pydbgen import pydbgen
myDB=pydbgen.pydb()
 
# 生成一个DataFrame
info = myDB.gen_dataframe(10,['name','phone','city','state'])

结果如下:

 

name

phone-number

city

state

0

Hannah Richard

810-859-7815

Irwinville

Louisiana

1

Ronald Berry

591-564-0585

Glen Ellen

Minnesota

2

Caitlin Barron

969-840-8580

Dubois

Oklahoma

3

Felicia Stephens

154-858-1233

Veedersburg

Alaska

4

Shelly Dennis

343-104-9365

Mattapex

Virginia

5

Nicholas Hill

992-239-1954

Moneta

Minnesota

6

Steve Bradshaw

164-081-7811

Ten Broeck

Colorado

7

Gail Johnston

155-259-9514

Wayan

Virginia

8

John Gray

409-892-4716

Darlington

Pennsylvania

9

Katherine Bautista

185-861-1677

McNab

Texas

假如现在我们要对city列进行进行拆分,按照空格拆分,转换成多行的数据,
第一步:拆分,生成多列

info_city = info['city'].str.split(' ', expand=True)

结果如下:

 

0

1

0

Irwinville

None

1

Glen

Ellen

2

Dubois

None

3

Veedersburg

None

4

Mattapex

None

5

Moneta

None

6

Ten

Broeck

7

Wayan

None

8

Darlington

None

9

McNab

None

可以看到已经将原始数据拆分成了2列,对于无法拆分的数据为None

第二步:行转列

info_city = info_city.stack()

结果如下:

 

 

 

0

0

Irwinville

1

0

Glen

 

1

Ellen

2

0

Dubois

3

0

Veedersburg

4

0

Mattapex

5

0

Moneta

6

0

Ten

 

1

Broeck

7

0

Wayan

8

0

Darlington

9

0

McNab

其中前面两列是索引,返回的是一个series,没有名字的series

第三步:重置索引,并命名(并删除多于的索引)

info_city = info_city.reset_index(level=1, drop=True)

结果如下:

 

 

0

Irwinville

1

Glen

1

Ellen

2

Dubois

3

Veedersburg

4

Mattapex

5

Moneta

6

Ten

6

Broeck

7

Wayan

8

Darlington

9

McNab

第四步:和原始数据合并

info_new = info.drop(['city'], axis=1).join(info_city)

结果如下:

 

name

phone-number

state

city

0

Hannah Richard

810-859-7815

Louisiana

Irwinville

1

Ronald Berry

591-564-0585

Minnesota

Glen

1

Ronald Berry

591-564-0585

Minnesota

Ellen

2

Caitlin Barron

969-840-8580

Oklahoma

Dubois

3

Felicia Stephens

154-858-1233

Alaska

Veedersburg

4

Shelly Dennis

343-104-9365

Virginia

Mattapex

5

Nicholas Hill

992-239-1954

Minnesota

Moneta

6

Steve Bradshaw

164-081-7811

Colorado

Ten

6

Steve Bradshaw

164-081-7811

Colorado

Broeck

7

Gail Johnston

155-259-9514

Virginia

Wayan

8

John Gray

409-892-4716

Pennsylvania

Darlington

9

Katherine Bautista

185-861-1677

Texas

McNab

需要特别注意的是,需要使用原始的连接新生成的,因为新生成的是一个series没有join方法,也可以通过将生成的series通过to_frame方法转换成DataFrame,这样就没有什么差异了

写了这么多,记住下面的就行了:

info.drop([‘city'], axis=1).join(info[‘city'].str.split(' ', expand=True).stack().reset_index(level=1, drop=True).rename(‘city'))

如果原数据中已经是list了,可以将info[‘city'].str.split(' ', expand=True)这部分替换成info[‘city'].apply(lambda x: pd.Series(x)),就可以达到相同的目的。

以上这篇Pandas实现一列数据分隔为两列就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持服务器之家。

原文链接:https://blog.csdn.net/Dennis_Shaw/article/details/96136723

标签:

相关文章

热门资讯

2022年最旺的微信头像大全 微信头像2022年最新版图片
2022年最旺的微信头像大全 微信头像2022年最新版图片 2022-01-10
蜘蛛侠3英雄无归3正片免费播放 蜘蛛侠3在线观看免费高清完整
蜘蛛侠3英雄无归3正片免费播放 蜘蛛侠3在线观看免费高清完整 2021-08-24
背刺什么意思 网络词语背刺是什么梗
背刺什么意思 网络词语背刺是什么梗 2020-05-22
yue是什么意思 网络流行语yue了是什么梗
yue是什么意思 网络流行语yue了是什么梗 2020-10-11
暖暖日本高清免费中文 暖暖在线观看免费完整版韩国
暖暖日本高清免费中文 暖暖在线观看免费完整版韩国 2021-05-08
返回顶部