Как назвать столбцы и строки матрицы с именами, встроенными в python?

Question

Мне нужно назвать столбцы и строки матрицы с именами столбцов, включенными в матрицу:

Пример:

M = ['A:1' 'B:0' 'C:0', 'A:0' 'B:1' 'C:0', 'A:0' 'B:0' 'C:1'] 

to:

M = ['1' 'A' 'B' 'C','2' '1' '0' '0','3' '0' '1' '0', '4' '0' '0' '1']

Очень похоже на формат Excel, где столбцы помечены переменными, включенными в матрицу, а строки пронумерованы.

Этот код следует использовать позже при анализе больших данных

Answer 1

Библиотека pandas расширяет массив массивов, предоставляя вам представление данных (DataFrame), которое ближе к Excel.

import pandas as pd
df = pd.DataFrame({'A': [1, 0, 0],
                   'B': [0, 1, 0],
                   'C': [0, 0, 1]})
print(df)

Вывод:

   A  B  C
0  1  0  0
1  0  1  0
2  0  0  1

Также имеется функция read_excel() и метод DataFrame.to_excel().

Для извлечения базового массива numpy из DataFrameиспользуйте атрибут .values:

print(df.values)

Вывод:

[[1 0 0]
 [0 1 0]
 [0 0 1]]