我遵循的是简短的tutorial on LinkedIn regarding multiindexed pandas dataframes,在这里我无法重现看似非常基本的操作(于3:00)。您无需观看视频即可解决问题。
以下使用seaborn数据集的代码片段将显示我无法使用视频中显示的技术向多索引熊猫数据帧添加列,并且还在SO帖子here中进行了介绍。 / p>
我们在这里:
import pandas as pd
import seaborn as sns
flights = sns.load_dataset('flights')
flights.head()
flights_indexed = flights.set_index(['year', 'month'])
flights_unstack = flights_indexed.unstack()
print(flights_unstack)
输出:
passengers
month January February March April May June July August September October November December
year
1949 112 118 132 129 121 135 148 148 136 119 104 118
1950 115 126 141 135 125 149 170 170 158 133 114 140
1951 145 150 178 163 172 178 199 199 184 162 146 166
1952 171 180 193 181 183 218 230 242 209 191 172 194
1953 196 196 236 235 229 243 264 272 237 211 180 201
1954 204 188 235 227 234 264 302 293 259 229 203 229
1955 242 233 267 269 270 315 364 347 312 274 237 278
1956 284 277 317 313 318 374 413 405 355 306 305 336
1957 315 301 356 348 355 422 465 467 404 347 310 337
1958 340 318 362 348 363 435 491 505 404 359 362 405
1959 360 342 406 396 420 472 548 559 463 407 362 405
1960 417 391 419 461 472 535 622 606 508 461 390 432
现在我想添加一列,以显示使用每年每年的总金额
flights_unstack.sum(axis = 1)
输出:
year
1949 1520
1950 1676
1951 2042
1952 2364
1953 2700
1954 2867
1955 3408
1956 3939
1957 4421
1958 4572
1959 5140
1960 5714
上面提到的两个消息源通过使用简单的方法演示了这一点:
flights_unstack['passengers', 'total'] = flights_unstack.sum(axis = 1)
此处,“总计”应在现有索引下显示为新列。 但我明白了:
TypeError:无法将项目插入尚不存在的类别CategoricalIndex中
我正在使用Python 3,2015年视频中的作者也是如此。
这是怎么回事?
我仅使用上述系列中的值进行了很多尝试,以及将数据bot重塑,转置,合并和连接为pd.Series和pd.DataFrame。并重置索引。我可能忽略了一些重要的细节,现在希望大家提出一些建议。
编辑1-这是jezrael的第一个建议之后的尝试:
import pandas as pd
import seaborn as sns
flights = sns.load_dataset('flights')
flights.head()
flights_indexed = flights.set_index(['year', 'month'])
flights_unstack = flights_indexed['passengers'].unstack()
flights_unstack['total'] = flights_unstack.sum(axis = 1)
输出:
TypeError:无法将项目插入尚不存在的类别CategoricalIndex中
答案 0 :(得分:2)
我知道这有点晚了,但是我在课程的“常见问题解答”部分找到了您问题的答案。它是这样的:
“问:熊猫分类数据有什么问题?
A。从0.6版开始,seaborn.load_dataset将某些列转换为Pandas分类数据(请参见http://pandas.pydata.org/pandas-docs/stable/categorical.html)。这在“熊猫简介/使用多级索引”中使用的“航班” DataFrame的处理中产生了问题。为避免此问题,您可以直接使用Pandas加载数据集:
航班= pd.read_csv('https://raw.githubusercontent.com/mwaskom/seaborn-data/master/flights.csv')“
我希望这会有所帮助。
答案 1 :(得分:1)
更改:
flights_unstack = flights_indexed.unstack()
收件人:
flights_unstack = flights_indexed['passengers'].unstack()
用于删除列中的Multiindex
。
最后add_categories
用新的列名是必需的:
flights_unstack.columns = flights_unstack.columns.add_categories(['total'])
flights_unstack['total'] = flights_unstack.sum(axis = 1)
print (df)
January February March April May June July August September \
month
1949 112 118 132 129 121 135 148 148 136
1950 115 126 141 135 125 149 170 170 158
1951 145 150 178 163 172 178 199 199 184
1952 171 180 193 181 183 218 230 242 209
1953 196 196 236 235 229 243 264 272 237
1954 204 188 235 227 234 264 302 293 259
1955 242 233 267 269 270 315 364 347 312
1956 284 277 317 313 318 374 413 405 355
1957 315 301 356 348 355 422 465 467 404
1958 340 318 362 348 363 435 491 505 404
1959 360 342 406 396 420 472 548 559 463
1960 417 391 419 461 472 535 622 606 508
October November December total
month
1949 119 104 118 1520
1950 133 114 140 1676
1951 162 146 166 2042
1952 191 172 194 2364
1953 211 180 201 2700
1954 229 203 229 2867
1955 274 237 278 3408
1956 306 305 336 4003
1957 347 310 337 4427
1958 359 362 405 4692
1959 407 362 405 5140
1960 461 390 432 5714
设置:
import pandas as pd
temp=u"""month;January;February;March;April;May;June;July;August;September;October;November;December
1949;112;118;132;129;121;135;148;148;136;119;104;118
1950;115;126;141;135;125;149;170;170;158;133;114;140
1951;145;150;178;163;172;178;199;199;184;162;146;166
1952;171;180;193;181;183;218;230;242;209;191;172;194
1953;196;196;236;235;229;243;264;272;237;211;180;201
1954;204;188;235;227;234;264;302;293;259;229;203;229
1955;242;233;267;269;270;315;364;347;312;274;237;278
1956;284;277;317;313;318;374;413;405;355;306;305;336
1957;315;301;356;348;355;422;465;467;404;347;310;337
1958;340;318;362;348;363;435;491;505;404;359;362;405
1959;360;342;406;396;420;472;548;559;463;407;362;405
1960;417;391;419;461;472;535;622;606;508;461;390;432"""
#after testing replace 'pd.compat.StringIO(temp)' to 'filename.csv'
df = pd.read_csv(pd.compat.StringIO(temp), sep=";", index_col=[0])
print (df)
df.columns = pd.CategoricalIndex(df.columns)
df.columns = df.columns.add_categories(['total'])
df['total'] = df.sum(axis = 1)
答案 2 :(得分:0)