Welcome to OGeek Q&A Community for programmer and developer-Open, Learning and Share
Welcome To Ask or Share your Answers For Others

Categories

0 votes
144 views
in Technique[技术] by (71.8m points)

python - Indexing single dimensional Axes array with two indices


与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome To Ask or Share your Answers For Others

1 Reply

0 votes
by (71.8m points)

If you want to use nan as the workaround, you can try this:

def fire_m(x, y, i, j=np.nan):
    if j == np.nan:
        ax[i].plot(x, y)
    else:
        ax[i,j].plot(x, y)

Another option is to use numpy.ravel_multi_index. This will convert the i, j-coordinate to a single index in a flattened version of the array.

def fire_m(x, y, i, j):
    n = np.ravel_multi_index([i, j], ax.shape)
    ax.flatten()[n].plot(x, y)

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
OGeek|极客中国-欢迎来到极客的世界,一个免费开放的程序员编程交流平台!开放,进步,分享!让技术改变生活,让极客改变未来! Welcome to OGeek Q&A Community for programmer and developer-Open, Learning and Share
Click Here to Ask a Question

...