如何使用元组列表中的索引创建 pandas DataFrame

Question

使用记录中的索引创建 pandas DataFrame 的最佳方法是什么。 这是我的示例：

sales = [('Jones LLC', 150, 200, 50),
     ('Alpha Co', 200, 210, 90),
     ('Blue Inc', 140, 215, 95)]
labels = ['account', 'Jan', 'Feb', 'Mar']
df = pd.DataFrame.from_records(sales, columns=labels)

我需要“帐户”作为此处的索引（不是列） 谢谢

Answer 1

最简单的就是set_index:

df = pd.DataFrame.from_records(sales, columns=labels).set_index('account')
print (df)
           Jan  Feb  Mar
account                 
Jones LLC  150  200   50
Alpha Co   200  210   90
Blue Inc   140  215   95

或者通过列表推导选择：

labels = [ 'Jan', 'Feb', 'Mar']
idx = [x[0] for x in sales]
data = [x[1:] for x in sales]

df = pd.DataFrame.from_records(data, columns=labels, index=idx)
print (df)
           Jan  Feb  Mar
Jones LLC  150  200   50
Alpha Co   200  210   90
Blue Inc   140  215   95