[英]Setting values with multiindex in pandas
關於SO的問題已經有幾個問題,尤其是這個問題 ,但是沒有一個答案似乎對我有用,並且很多文檔鏈接(特別是關於lexsorting)都被打破了,所以我會問另一個。
我正在嘗試做某事(看似)非常簡單。 請考慮以下MultiIndexed Dataframe:
import pandas as pd; import random
arrays = [['bar', 'bar', 'baz', 'baz', 'foo', 'foo', 'qux', 'qux'],
['one', 'two', 'one', 'two', 'one', 'two', 'one', 'two']]
tuples = list(zip(*arrays))
index = pd.MultiIndex.from_tuples(tuples, names=['first', 'second'])
df = pd.concat([pd.Series(np.random.randn(8), index=index), pd.Series(np.random.randn(8), index=index)], axis=1)
現在我想將第0
列中的所有值設置為某個值(例如np.NaN
)以用於第one
類中的觀察。 我失敗了:
df.loc(axis=0)[:, "one"][0] = 1 # setting with copy warning
和
df.loc(axis=0)[:, "one", 0] = 1
這或者產生關於鍵的長度超過索引長度的警告,或者關於缺少lexsorting到足夠深度的警告。
這樣做的正確方法是什么?
我認為您可以使用帶有元組的loc
來選擇MultiIndex
,使用0
來選擇列:
import pandas as pd;
import random
arrays = [['bar', 'bar', 'baz', 'baz', 'foo', 'foo', 'qux', 'qux'],
['one', 'two', 'one', 'two', 'one', 'two', 'one', 'two']]
#add for testing
np.random.seed(0)
tuples = list(zip(*arrays))
index = pd.MultiIndex.from_tuples(tuples, names=['first', 'second'])
df = pd.concat([pd.Series(np.random.randn(8), index=index), pd.Series(np.random.randn(8), index=index)], axis=1)
print df
0 1
first second
bar one 1.764052 -0.103219
two 0.400157 0.410599
baz one 0.978738 0.144044
two 2.240893 1.454274
foo one 1.867558 0.761038
two -0.977278 0.121675
qux one 0.950088 0.443863
two -0.151357 0.333674
df.loc[('bar', "one"), 0] = 1
print df
0 1
first second
bar one 1.000000 -0.103219
two 0.400157 0.410599
baz one 0.978738 0.144044
two 2.240893 1.454274
foo one 1.867558 0.761038
two -0.977278 0.121675
qux one 0.950088 0.443863
two -0.151357 0.333674
如果需要將second
級中的所有行設置為值one
使用slice(None)
:
df.loc[(slice(None), "one"), 0] = 1
print df
0 1
first second
bar one 1.000000 -0.103219
two 0.400157 0.410599
baz one 1.000000 0.144044
two 2.240893 1.454274
foo one 1.000000 0.761038
two -0.977278 0.121675
qux one 1.000000 0.443863
two -0.151357 0.333674
文件 。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.