代码之家 › 专栏 › 技术社区 › TylerNG

使用pandas生成矩阵

count matrix pandas

TylerNG · 技术社区 · 8 年前

我想使用pandas为数据df生成一个矩阵,逻辑如下:

按id分组

低:中上:结束

对于第1天:计算if(如果级别有中间和结尾,并且如果日期==1)

对于第2天:计算if(如果级别有中间和结尾,并且如果日期==2)

开始:中到新

对于第1天:计算if(如果级别有Mid和New,并且如果日期==1)

对于第2天:计算if(如果级别有Mid和New,并且如果日期==2)

df=pd。数据帧({'Id':[111111 22233334445555555566666],'Level':['End','Mid','End','Mid','New','End','New','Mid','New','Mid','day':['',3','','',2,3',3,4',2]})

Id |级别|天

111 |结束|

111 |中| 3

222 |结束|

333 |结束|

333 |中| 2

444 |新| 3

555 |结束|

555 |新| 3

555 |中| 4

666 |新|

666 |中| 2

矩阵如下所示:

低顶第1天第2天第3天第4天

中端0 1 1 0

中新0 1 0 1

新结束0 0 1 0

新建Mid 0 0 1

非常感谢。非常感谢。

1 回复 | 直到 8 年前

00__00__00 8 年前

从数据帧开始

 # all the combination of Levels
level_combos=[c for c in itertools.combinations(df['Level'].unique().tolist(), 2)]
 # create output and fill with zeros
df_output=pd.DataFrame(0,index=level_combos,columns=range(4))

可能效率不高,但应该可以

for g in df.groupby(['Id']): # group by ID
    # combination of levels for this ID
    level_combos_this_id=[c for c in itertools.combinations(g[1]['Level'].unique().tolist(), 2)]


   # set to 1 the days present
    df_output.loc[level_combos_this_id,pd.to_numeric(g[1]['day']).dropna(inplace=True).values]=1

最后重命名列以获得所需的输出

df_output.columns=['day'+str(i+1) for i in range(4)]

推荐文章

Homer Jay Simpson · 添加列出现次数

1 年前

Tez · 数据帧-统计三列中具有相似(相同)变量的变量的频率

2 年前

swat · 在同一列上选择SQL语句

2 年前

Lisa · 如何使用Python找到一个表中哪些标记行位于或高于另一个表的某些点?

2 年前

Giampaolo Levorato · 计数Pandas列表中的连续元素

2 年前

Elijah · 计算与其他列值关联的列中的值集[重复]

2 年前

Chulho Chang · 计算一行中特定字符的数量(不在范围内)

2 年前

Clark Hung · 如果主类别相同,如何统计所有子类别,但只统计一次

2 年前

Rajesh Ahir · 如何在pandas数据帧的列中找到连续的null(NaN)?

2 年前

TarJae · 如何在一次运行中对多个列计数使用计数

2 年前