python 值列表，如何？

Question

我不知道该怎么说

但我有一个 say 的列表

l = ['a','b','c','d','e']  

这是排序和索引所以如果我做 l[0] => 'a'

我使用 np.eye(5) 得到对角线 1

array([[ 1., 0., 0., 0., 0.], 
[ 0., 1., 0., 0., 0.], 
[ 0., 0., 1., 0., 0.], 
[ 0., 0., 0., 1., 0.], 
[ 0., 0., 0., 0., 1.]])

但我的目标是得到类似..

array([[ a., 0., 0., 0., 0.], 
[ 0., b., 0., 0., 0.], 
[ 0., 0., c., 0., 0.], 
[ 0., 0., 0., d., 0.], 
[ 0., 0., 0., 0., e.]])

=============更新===================

np 这里是 numpy 库，不好意思

将 numpy 导入为 np

Answer 1

所以你想用向量创建一个对角线数组，对吧？那是 diag :

>>> l = ['a','b','c','d','e']  
>>> np.diag(l)
array([['a', '', '', '', ''],
       ['', 'b', '', '', ''],
       ['', '', 'c', '', ''],
       ['', '', '', 'd', ''],
       ['', '', '', '', 'e']],
      dtype='|S1')

或者，如果这些是变量而不是字符串:

>>> a, b, c, d, e = 1., 2., 3., 4., 5.
>>> l = [a, b, c, d, e]
>>> np.diag(l)
array([[ 1.,  0.,  0.,  0.,  0.],
       [ 0.,  2.,  0.,  0.,  0.],
       [ 0.,  0.,  3.,  0.,  0.],
       [ 0.,  0.,  0.,  4.,  0.],
       [ 0.,  0.,  0.,  0.,  5.]])