描述

将数组前 $n$ 个元素设为类别 $A$，其余设为类别 $B$。

格式

输入

两行：第一行是元素列表，第二行是分割点 $n$

输出

打印分类数组。

样例

[1,2,3,4,5,6]
3

['A' 'A' 'A' 'B' 'B' 'B']

提示

$\texttt{np.where}$(np.arange(len(Z)) < $n$, 'A', 'B')。