Поиск массива Numery для индексов нескольких значений [duplicate]

<?php
      $username = $_POST['username'];
       $password = $_POST['password'];

     $result = mysql_query("SELECT * FROM Users WHERE UserName LIKE '".mysql_real_escape_string($username)."'")or die(mysql_error());
while($row=mysql_fetch_array($result))
  {
 echo $row['FirstName'];
 }
 ?>
10
задан Divakar 12 November 2015 в 19:58
поделиться

3 ответа

Вы можете использовать np.in1d с np.nonzero -

np.nonzero(np.in1d(A,B))[0]

Вы также можете использовать np.searchsorted , если вы хотите сохранить порядок -

np.searchsorted(A,B)

Для общего случая, когда A & amp; B - несортированные массивы, вы можете ввести параметр sorter в np.searchsorted, например:

sort_idx = A.argsort()
out = sort_idx[np.searchsorted(A,B,sorter = sort_idx)]

Я бы добавил в свою любимую broadcasting тоже в миксе, чтобы решить общий случай -

np.nonzero(B[:,None] == A)[1]

Прогон образца -

In [125]: A
Out[125]: array([ 7,  5,  1,  6, 10,  9,  8])

In [126]: B
Out[126]: array([ 1, 10,  7])

In [127]: sort_idx = A.argsort()

In [128]: sort_idx[np.searchsorted(A,B,sorter = sort_idx)]
Out[128]: array([2, 4, 0])

In [129]: np.nonzero(B[:,None] == A)[1]
Out[129]: array([2, 4, 0])
11
ответ дан Divakar 24 August 2018 в 07:55
поделиться

Вы пробовали searchsorted?

A = np.array([1,2,3,4,5,6,7,8,9,10])
B = np.array([1,7,10])

A.searchsorted(B)
# array([0, 6, 9])
4
ответ дан Bi Rico 24 August 2018 в 07:55
поделиться

Просто для полноты: если значения в A не отрицательны и достаточно малы:

lookup = np.empty((np.max(A) + 1), dtype=int)
lookup[A] = np.arange(len(A))
indices  = lookup[B]
1
ответ дан Paul Panzer 24 August 2018 в 07:55
поделиться
Другие вопросы по тегам:

Похожие вопросы: