midway.py
a19ef641815etranslation_to_python

midway.py
View Options

			import numpy as np

			def midway(u_1, u_2):
			u_midway_1 = np.zeros(u_1.shape)
			u_midway_2 = np.zeros(u_2.shape)

			index_u_1 = np.argsort(u_1, axis=None)
			index_u_2 = np.argsort(u_2, axis=None)
			u_sort_1 = u_1.flatten()[index_u_1]
			u_sort_2 = u_2.flatten()[index_u_2]

			u_midway_1[np.unravel_index(index_u_1, u_1.shape)] = (u_sort_1 + u_sort_2) / 2
			u_midway_2[np.unravel_index(index_u_2, u_2.shape)] = (u_sort_1 + u_sort_2) / 2

			u_midway_1 = u_midway_1.reshape(u_1.shape)
			u_midway_2 = u_midway_2.reshape(u_2.shape)

			return [u_midway_1, u_midway_2]