for index,row in dr.iterrows():
if(dr['Level :'][index].day == today.day or dr['Level :'][index].day == tomo.day ):
dr1 = row
else:
dr2 = row
实际上,我是这个熊猫概念的新手。我根据条件将数据从dr
数据帧复制到dr2
数据帧。 dr
数据框中的这些行将被复制为数据框dr2
中的列。因此,我应该在代码中进行哪些更改,以保留与dr2
数据框相同的dr
数据框形状。
答案 0 :(得分:1)
您可以选择满足条件的行indeces。然后使用它们来分割初始数据帧(我想你也想要reset_index()):
dr["day"] = dr["Level :"].apply(lambda x: x.day)
ix = dr["day"].isin([today.day, tomo.day])
dr1 = dr.loc[ix].reset_index(drop=True)
dr2 = dr.loc[~ix].reset_index(drop=True)