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

Question

Моя функция выводит список, например, когда я типа:

My_function('TV', 'TV_Screen') 

выводит следующее:

['TV', 1, 'TV_Screen', 0.04, 'True'] 

Теперь мой телевизор сделан из нескольких таких как динамик, трансформатор и т. д., я могу продолжать выполнять свою функцию для каждой части и, например, менять «TV_Screen» для «TV_Speaker» или «TV_transformer» и т. д.

Альтернативой является создание лис т со всей части, такие как:

TV_parts = ['TV_Screen', 'TV_Speaker', 'TV_transformer'] 

То, что я пытаюсь получить это кадр панды данных с 5 столбцов (потому что мои функции выходов 5 переменных, смотрите выше в разделе «выводит следующее:») и в этом случае 3 ряда (по одному для «TV_Screen», «TV_Speaker» и «TV_transformer»). В принципе я хочу следующее быть в кадре данных:

['TV', 1, 'TV_Screen', 0.04, 'True'] 
['TV', 9, 'TV_Speaker', 0.56, 'True'] 
['TV', 3, 'TV_transformer', 0.80, 'False'] 

Я знаю, что нужно для цикла где-то, но я не уверен, как создать этот фрейм данных. Не могли бы вы помочь? (Я могу изменить вывод моей функции как pd.Series или что-то еще, что будет работать лучше).

Спасибо!

Answer 1

Вы можете сделать это следующим образом:

def My_function(part): 
    # prepare result 
    result = ['TV', 1, part, 0.04, 'True'] # for testing 
    return result 

TV_parts = ['TV_Screen', 'TV_Speaker', 'TV_transformer'] 
df = pd.DataFrame([My_function(part) for part in TV_parts]) 

>>> df 

    0 1    2  3  4 
0 TV 1  TV_Screen 0.04 True 
1 TV 1  TV_Speaker 0.04 True 
2 TV 1 TV_transformer 0.04 True 

Answer 2

Если у вас есть много массивов, может быть стоит их преобразования в Numpy матрицу первого, а затем преобразовывать их в dataframe.

import pandas as pd 
import numpy as np 

a = ['TV', 1, 'TV_Screen', 0.04, 'True'] 
b = ['TV', 9, 'TV_Speaker', 0.56, 'True'] 
c = ['TV', 3, 'TV_transformer', 0.80, 'False'] 

matrix = np.matrix([a,b,c]) 

df = pd.DataFrame(data=matrix)