Окрашенная сборка производится в Visual Studio

Прежде всего, похоже, что вы пропустили запятую в строке 6:

I = sp.array([0,1,4])

Во-вторых, я ожидал бы, что значение 4 в массиве I поднимет IndexError, так как оба и b имеет максимальный размер 4. Я подозреваю, что вам может понадобиться:

I = sp.array([0,1,3])

Выполнение этих изменений запускает программу для меня, и я получил b как:

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

Я подозреваю, что вы хотите.