Python превращает серию сериалов в DataFrames

Question

Создав панду DataFrame из метода groupby и apply, я сейчас пытаюсь извлечь подмножество этого DataFrame в новый DataFrame. Я создал функцию, которая извлекает данные, но мне нужно, но она возвращает данные в виде серии последовательностей. К сожалению, я не могу просто напрямую превратить эти серии в DataFrames, используя dicts, так как каждая серия - это строка, а не столбец. Я разместил код для функции. Ниже большое спасибо за любую помощь, вы можете иметь.

    def sel(df):
      for i in range(len(df)):
        if len(df.iloc[i][4].split()) == 3:
          return df.iloc[i][1:4]

Затем я запускаю эту функцию и присваиваю ее переменной.

Answer 1

Похоже, вы хотите взять столбцы с 1 по 3 из строк, которые имеют три слова в столбце 4, правильно? Вы можете сделать это без каких-либо петель.

df = pd.DataFrame([[0, 1,  2,  3, "a b c"],
                   [4, 5,  6,  7, "a b"],
                   [8, 9, 10, 11, "d e f"]])
df[df[4].str.split().str.len() == 3].iloc[:, 1:4]
#   1   2   3
#0  1   2   3
#2  9  10  11