使用可能的最简单索引在python pandas中转置一列

Question

我有以下数据（data_current）：

import pandas as pd
import numpy as np

data_current=pd.DataFrame({'medicine':['green tea','fried tomatoes','meditation','meditation'],'disease':['acne','hypertension', 'cancer','lupus']})
data_current

我想做的是移置其中一列，这样我就不必为同一药物和不同疾病设置多行，而为每种药物分配一行，并针对疾病设置几列。保持索引尽可能简单也很重要，例如0,1,2 ...，即我不想将'medicines'分配为索引列，因为我将其合并到其他键上。所以，我需要data_needed

data_needed=pd.DataFrame({'medicine':['green tea','fried tomatoes','meditation'],'disease_1':['acne','hypertension','cancer'], 'disease_2':['np.nan','np.nan','lupus']})
data_needed

Answer 1

我在想你要一个数据透视表。检查此链接以获取更多信息-> http://pandas.pydata.org/pandas-docs/stable/reshaping.html

您发现此输出可接受吗？

data_current.pivot(index='medicine', columns='disease', values='disease')