Использование строк пути в качестве индекса в пандах

Я пытаюсь создать фрейм данных с путями к файлам в качестве индекса:

import os
import pandas as pd
pathnames = []
for i in range(5):
    pathnames.append(os.path.join('a',str(i)))

print(pathnames)
df = pd.DataFrame(index = pathnames)
df[pathnames[0]]

При использовании этого примера я получаю ошибку ключа как pathnames[0] интерпретируется как a\\0 (как при печати), но не как a\0. Я ожидаю, что проблема будет скрыта в форматировании необработанной строки. Может кто-нибудь объяснить и помочь? протестированная версия pandas: 0.24.2 и 1.1.0

1 ответ

Используйте loc или iloc

import os
import pandas as pd
pathnames = []
for i in range(5):
    pathnames.append(os.path.join('a',str(i)))

print(pathnames)
df = pd.DataFrame(index = pathnames)
print(df.loc['a/0'])

Другие вопросы по тегам