Python的切片与异步，操场上有10x10的同学，能自己选指定不规则的列数吗？

Question

# 操场上有10x10的同学

z ='''
同学第1排：a,b,c,d,e,f
同学第2排：1,2,3,4,5,6
同学第3排：你,我,他,她,它,她们
'''

# 比如说：我选c,d,e列，我这里是选随机数列
input() 
# 任意249,125,978,432…
。
print

Answer 1

import numpy as np

# 操场上有10x10的同学
z ='''
同学第1排：a,b,c,d,e,f
同学第2排：1,2,3,4,5,6
同学第3排：你,我,他,她,它,她们
'''
l = [i.split('：')[1].split(',') for i in z.strip().split()]
ar = np.array(l)
col = list(map(int, list(input(">>>"))))

print(ar[:, col])


--result
>>>234
[['c' 'd' 'e']
 ['3' '4' '5']
 ['他' '她' '它']]