python, issue with for loop and plotting

Question

import numpy as np
import time
import matplotlib
import matplotlib.pyplot as plt
A=np.array([[0,0,0,0,0,0,0,0,1,0],[0,0,1,0,1,0,0,