копировать данные из одного кадра данных в другой

Question

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 dataframe в dr2 dataframe в зависимости от условия.Эти строки в dr dataframe копируются как столбцы в dataframe dr2.Итак, что я должен изменить в своем коде, чтобы сохранить форму dr2 dataframe такой же, как dr dataframe.

Answer 1

Вы можете выбрать число строк, которые удовлетворяют условию.Затем используйте их для разделения исходного кадра данных (я думаю, вы также хотите 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)