Преобразование DataFrame в многомерный массив с именами столбцов DataFrame - PullRequest
0 голосов
/ 28 февраля 2019

Ниже приведен кадр данных, на который я хочу воздействовать:

df = pd.DataFrame({'A': [1,1,1],
                   'B': [2,2,3],
                   'C': [4,5,4]})

Каждая строка df создает уникальный ключ.Цель состоит в том, чтобы создать следующий список многомерных массивов:

parameter = [[['A', 1],['B', 2], ['C', 4]],
             [['A', 1],['B', 2], ['C', 5]],
             [['A', 1],['B', 3], ['C', 4]]]

Проблема связана с этим вопросом , где я должен перебирать parameter, но вместо того, чтобы предоставлять их вручнуюк моей функции я должен поместить все parameter из df (строк) в список.

1 Ответ

0 голосов
/ 28 февраля 2019

Вы можете использовать следующее понимание списка, которое упаковывает значения в каждой строке в столбцы кадра данных:

from itertools import repeat

[list(map(list,zip(cols, i))) for cols, i in zip(df.values.tolist(), repeat(df.columns))]

[[[1, 'A'], [2, 'B'], [4, 'C']],
 [[1, 'A'], [2, 'B'], [5, 'C']],
 [[1, 'A'], [3, 'B'], [4, 'C']]]
...