Channel Routingをpythonで実装したい

Question

前提・実現したいこと

channel routing_figure1

質問失礼します。
上のリンクのfigure1の画像のように、回路設計のChannel Routingをpythonで実装したいと考えています。

Channel Routingでは、図のようにtopとbottomに指定されたピンを同じ数字同士で接続する必要があります。そこで、horizontal segmentの長さを決めるためのプログラムを考えています。

top = [1 ,0, 3, 1, 4, 2, 3, 2]
bottom = [6, 4, 6, 6, 3, 0, 5, 5]

のように入力データを指定してピン1~6の長さをどのようにpythonで書けば良いかのアドバイスをいただきたいと思っております。
4の場合ですと、top[4]とbottom[1]の距離がわかればよいです。

よろしくお願いします。

Answer 1

numpyのwhereを使うのが簡単でしょう。

>>> top = [1 ,0, 3, 1, 4, 2, 3, 2]
>>> bottom = [6, 4, 6, 6, 3, 0, 5, 5]
>>>
>>> import numpy as np
>>>
>>> def h_seg(top, bottom, num):
...     top_index = np.where(np.array(top) == num)
...     bottom_index = np.where(np.array(bottom) == num)
...     concat = np.concatenate([top_index, bottom_index], axis=1)
...     return concat.max() - concat.min()
...
>>> print(h_seg(top, bottom, 4))
3
>>> print(h_seg(top, bottom, 3))
4
>>> print(h_seg(top, bottom, 6))
3

whereについては、NumPyで条件に応じた処理を行うnp.whereの使い方をお読みください。

Comments

[iface]

回答ありがとうございます。
whereの使い方を初めて知りました。
こちらの作成していただいたプログラムを参考にさせていただきます。

[iface]

>>> concat = np.concatenate([top_index, bottom_index], axis=1)

質問失礼します。
こちらの構文を詳しく説明していただけますでしょうか。