子集pandas数据帧并保留原始大小

Question

我正在尝试对数据帧进行子集化,但希望新数据帧具有相同大小的原始数据帧.
附加输入,输出和预期输出.

df_input = pd.DataFrame([[1,2,3,4,5], [2,1,4,7,6], [5,6,3,7,0]], columns=["A", "B","C","D","E"])

df_output=pd.DataFrame(df_input.iloc[1:2,:])

df_expected_output=pd.DataFrame([[0,0,0,0,0], [2,1,4,7,6], [0,0,0,0,0]], columns=["A", "B","C","D","E"])  

请建议前进的方向.

Answer 1

将子集返回到原始子集后设置索引reindex.这将设置新行的所有值NaN,您可以将其替换为0 via fillna.由于NaN是floaT型,你可以一切转换回int用astype.

 df_input.iloc[1:2,:].reindex(df_input.index).fillna(0).astype(int)