Нужно добавить имена столбцов в массив numpy

Question

Я пытаюсь создать игру connect 4 с массивом 6/7 в python, и мне нужны заголовки столбцов, чтобы столбец 0 назывался a, столбец 2 назывался b и так далее.Цель этого состоит в том, чтобы инициировать ходы, набрав «a» (удаляет токен в первом столбце), «b» (удаляет токен во втором) и т. Д. .... Это мой код для создания массива

def clear_board():
     board = np.zeros((6,7))
     return board

Answer 1

Если вам нужны имена столбцов, самый простой способ - использовать pandas массив данных вместо numpy массива:

import pandas as pd 

def clear_board():
    board = pd.DataFrame(np.zeros((6,7)),columns=list('ABCDEFG'))
    return board

>>> clear_board()
     A    B    C    D    E    F    G
0  0.0  0.0  0.0  0.0  0.0  0.0  0.0
1  0.0  0.0  0.0  0.0  0.0  0.0  0.0
2  0.0  0.0  0.0  0.0  0.0  0.0  0.0
3  0.0  0.0  0.0  0.0  0.0  0.0  0.0
4  0.0  0.0  0.0  0.0  0.0  0.0  0.0
5  0.0  0.0  0.0  0.0  0.0  0.0  0.0

Помимо этого, взгляните на опции, представленные в этот ответ