Python: создание фрейма данных с «неровными» записями столбцов

Question

Я пытаюсь создать фрейм данных, где длины столбцов не равны. Как я могу это сделать? Я пытался использовать groupby. Но я думаю, что это не правильный путь.

import pandas as pd

data = {'filename':['file1','file1'], 'variables':['a','b']}
df = pd.DataFrame(data)

grouped = df.groupby('filename')
print(grouped.get_group('file1'))

Выше приведен мой пример кода. Выход которого:

Что я могу сделать, чтобы иметь только одну запись 'file1' в 'filename'? В конце концов мне нужно записать это в CSV-файл.

Спасибо

Answer 1

Если у вас есть только одна запись в столбце, другой будет NaN. Таким образом, вы можете просто отфильтровать NaN, выполнив что-то вроде df = df.at[df["filename"].notnull()]