在MultiIndex DataFrame中添加一个空行。

3
正如标题所示,我想向我的MultiIndex DataFrame中添加一个空行。第一级索引需要有一个定义的索引值,第二级索引需要是np.nan。列中的值需要是np.nan。
考虑以下内容:
import pandas as pd
import numpy as np

iterables = [['foo'], ['r_1', 'r_2', 'r_3']]
idx = pd.MultiIndex.from_product(iterables, names=['idx_1', 'idx_2'])
data = [(1, 2, 3), (4, 5, 6), (7, 8, 9)]
df = pd.DataFrame(data, idx, columns=['col_1', 'col_2', 'col_3'])
df
Out[93]:
             col_1  col_2  col_3
idx_1 idx_2                     
foo   r_1        1      2      3
      r_2        4      5      6
      r_3        7      8      9

如果这不是一个 MultiIndex,我通常会像这样添加一个 Series

s = pd.Series(
    [np.nan, np.nan, np.nan], 
    index=['col_1', 'col_2', 'col_3'], 
    name='bar'
)
df.append(s)
Out[95]:
            col_1  col_2  col_3
(foo, r_1)    1.0    2.0    3.0
(foo, r_2)    4.0    5.0    6.0
(foo, r_3)    7.0    8.0    9.0
bar           NaN    NaN    NaN

在这种情况下,我的MultiIndex被转换为元组。我不能在append方法中使用ignore_index=True,因为那会删除MultiIndex。我感觉我已经很接近了,但还差一点。
我的输出应该像这样:
# some magic
Out[96]:
             col_1  col_2  col_3
col_a col_b
foo   r_1    1.0    2.0    3.0
      r_2    4.0    5.0    6.0
      r_3    7.0    8.0    9.0
bar   NaN    NaN    NaN    NaN

(也可以接受第二级索引为)。

我该怎么做?

使用Python 3.6和Pandas 0.20.3。

1个回答

3
使用“扩展设定”(setting with enlargement)功能:了解更多
df.loc[('bar', ''), ['col_1', 'col_2', 'col_3']] = np.nan

或者在“name”中使用元组:
s = pd.Series(
    [np.nan, np.nan, np.nan], 
    index=['col_1', 'col_2', 'col_3'], 
    name=('bar', np.nan)
)

print (df.append(s))
             col_1  col_2  col_3
idx_1 idx_2                     
foo   r_1      1.0    2.0    3.0
      r_2      4.0    5.0    6.0
      r_3      7.0    8.0    9.0
bar   NaN      NaN    NaN    NaN

s = pd.Series(
    [np.nan, np.nan, np.nan], 
    index=['col_1', 'col_2', 'col_3'], 
    name=('bar', '')
)

print (df.append(s))
             col_1  col_2  col_3
idx_1 idx_2                     
foo   r_1      1.0    2.0    3.0
      r_2      4.0    5.0    6.0
      r_3      7.0    8.0    9.0
bar            NaN    NaN    NaN

1
啊,该死。我在 name 参数中使用了一个列表,然后 Series 报错了:Series.name 必须是可哈希类型。 - Jason Strimpel
正确的链接(编辑队列已满):https://pandas.pydata.org/pandas-docs/stable/user_guide/indexing.html#setting-with-enlargement。 - young_souvlaki

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接