Using np.where to find element from sub-arrays












4















I'm trying to find all elements in array X that match element on array Y using np.where() and the condition on where() function is comparing list (a) not one element. Please see the following code:



X = np.array([[0, 2], [2, 1], [1, 3], [5, 9], [6, 7], [4, 6]])
Y = np.array([1, 2, 3, 4, 4, 5])
a = [2, 3, 4]
matchedX = X[np.where(Y == a)]


I am expecting to get the result like this:



array([[2, 1],
[1, 3],
[5, 9],
[6, 7]])


but I got different results:



array(, shape=(0, 2), dtype=int64)


So, I need an alternative solution in a way that I can get the same elements if I do not know about the values of a? This line below give me the exact results that I want, but I do not know in previous the a values.



matchedX = X[np.where((Y == 2) | (Y==3) | (Y==4))]









share|improve this question



























    4















    I'm trying to find all elements in array X that match element on array Y using np.where() and the condition on where() function is comparing list (a) not one element. Please see the following code:



    X = np.array([[0, 2], [2, 1], [1, 3], [5, 9], [6, 7], [4, 6]])
    Y = np.array([1, 2, 3, 4, 4, 5])
    a = [2, 3, 4]
    matchedX = X[np.where(Y == a)]


    I am expecting to get the result like this:



    array([[2, 1],
    [1, 3],
    [5, 9],
    [6, 7]])


    but I got different results:



    array(, shape=(0, 2), dtype=int64)


    So, I need an alternative solution in a way that I can get the same elements if I do not know about the values of a? This line below give me the exact results that I want, but I do not know in previous the a values.



    matchedX = X[np.where((Y == 2) | (Y==3) | (Y==4))]









    share|improve this question

























      4












      4








      4








      I'm trying to find all elements in array X that match element on array Y using np.where() and the condition on where() function is comparing list (a) not one element. Please see the following code:



      X = np.array([[0, 2], [2, 1], [1, 3], [5, 9], [6, 7], [4, 6]])
      Y = np.array([1, 2, 3, 4, 4, 5])
      a = [2, 3, 4]
      matchedX = X[np.where(Y == a)]


      I am expecting to get the result like this:



      array([[2, 1],
      [1, 3],
      [5, 9],
      [6, 7]])


      but I got different results:



      array(, shape=(0, 2), dtype=int64)


      So, I need an alternative solution in a way that I can get the same elements if I do not know about the values of a? This line below give me the exact results that I want, but I do not know in previous the a values.



      matchedX = X[np.where((Y == 2) | (Y==3) | (Y==4))]









      share|improve this question














      I'm trying to find all elements in array X that match element on array Y using np.where() and the condition on where() function is comparing list (a) not one element. Please see the following code:



      X = np.array([[0, 2], [2, 1], [1, 3], [5, 9], [6, 7], [4, 6]])
      Y = np.array([1, 2, 3, 4, 4, 5])
      a = [2, 3, 4]
      matchedX = X[np.where(Y == a)]


      I am expecting to get the result like this:



      array([[2, 1],
      [1, 3],
      [5, 9],
      [6, 7]])


      but I got different results:



      array(, shape=(0, 2), dtype=int64)


      So, I need an alternative solution in a way that I can get the same elements if I do not know about the values of a? This line below give me the exact results that I want, but I do not know in previous the a values.



      matchedX = X[np.where((Y == 2) | (Y==3) | (Y==4))]






      python numpy






      share|improve this question













      share|improve this question











      share|improve this question




      share|improve this question










      asked Nov 17 '18 at 0:04









      Esraa MohamedEsraa Mohamed

      234




      234
























          2 Answers
          2






          active

          oldest

          votes


















          1














          You can use the set functions of numpy:



          X[np.where(np.isin(Y, a))]

          array([[2, 1],
          [1, 3],
          [5, 9],
          [6, 7]])





          share|improve this answer
























          • It works fine, thank you :)

            – Esraa Mohamed
            Nov 17 '18 at 0:20



















          1














          You can skip the np.where, which is redundant here, and just index using np.isin:



          X[np.isin(Y,a)]

          array([[2, 1],
          [1, 3],
          [5, 9],
          [6, 7]])


          This is because np.isin gives you a boolean array of where Y is in a:



          array([False,  True,  True,  True,  True, False])


          So by indexing with this array, it only selects those rows where True






          share|improve this answer





















          • 1





            It works fine, thank you :)

            – Esraa Mohamed
            Nov 17 '18 at 0:20











          Your Answer






          StackExchange.ifUsing("editor", function () {
          StackExchange.using("externalEditor", function () {
          StackExchange.using("snippets", function () {
          StackExchange.snippets.init();
          });
          });
          }, "code-snippets");

          StackExchange.ready(function() {
          var channelOptions = {
          tags: "".split(" "),
          id: "1"
          };
          initTagRenderer("".split(" "), "".split(" "), channelOptions);

          StackExchange.using("externalEditor", function() {
          // Have to fire editor after snippets, if snippets enabled
          if (StackExchange.settings.snippets.snippetsEnabled) {
          StackExchange.using("snippets", function() {
          createEditor();
          });
          }
          else {
          createEditor();
          }
          });

          function createEditor() {
          StackExchange.prepareEditor({
          heartbeatType: 'answer',
          autoActivateHeartbeat: false,
          convertImagesToLinks: true,
          noModals: true,
          showLowRepImageUploadWarning: true,
          reputationToPostImages: 10,
          bindNavPrevention: true,
          postfix: "",
          imageUploader: {
          brandingHtml: "Powered by u003ca class="icon-imgur-white" href="https://imgur.com/"u003eu003c/au003e",
          contentPolicyHtml: "User contributions licensed under u003ca href="https://creativecommons.org/licenses/by-sa/3.0/"u003ecc by-sa 3.0 with attribution requiredu003c/au003e u003ca href="https://stackoverflow.com/legal/content-policy"u003e(content policy)u003c/au003e",
          allowUrls: true
          },
          onDemand: true,
          discardSelector: ".discard-answer"
          ,immediatelyShowMarkdownHelp:true
          });


          }
          });














          draft saved

          draft discarded


















          StackExchange.ready(
          function () {
          StackExchange.openid.initPostLogin('.new-post-login', 'https%3a%2f%2fstackoverflow.com%2fquestions%2f53346940%2fusing-np-where-to-find-element-from-sub-arrays%23new-answer', 'question_page');
          }
          );

          Post as a guest















          Required, but never shown

























          2 Answers
          2






          active

          oldest

          votes








          2 Answers
          2






          active

          oldest

          votes









          active

          oldest

          votes






          active

          oldest

          votes









          1














          You can use the set functions of numpy:



          X[np.where(np.isin(Y, a))]

          array([[2, 1],
          [1, 3],
          [5, 9],
          [6, 7]])





          share|improve this answer
























          • It works fine, thank you :)

            – Esraa Mohamed
            Nov 17 '18 at 0:20
















          1














          You can use the set functions of numpy:



          X[np.where(np.isin(Y, a))]

          array([[2, 1],
          [1, 3],
          [5, 9],
          [6, 7]])





          share|improve this answer
























          • It works fine, thank you :)

            – Esraa Mohamed
            Nov 17 '18 at 0:20














          1












          1








          1







          You can use the set functions of numpy:



          X[np.where(np.isin(Y, a))]

          array([[2, 1],
          [1, 3],
          [5, 9],
          [6, 7]])





          share|improve this answer













          You can use the set functions of numpy:



          X[np.where(np.isin(Y, a))]

          array([[2, 1],
          [1, 3],
          [5, 9],
          [6, 7]])






          share|improve this answer












          share|improve this answer



          share|improve this answer










          answered Nov 17 '18 at 0:12









          Matthias OssadnikMatthias Ossadnik

          57427




          57427













          • It works fine, thank you :)

            – Esraa Mohamed
            Nov 17 '18 at 0:20



















          • It works fine, thank you :)

            – Esraa Mohamed
            Nov 17 '18 at 0:20

















          It works fine, thank you :)

          – Esraa Mohamed
          Nov 17 '18 at 0:20





          It works fine, thank you :)

          – Esraa Mohamed
          Nov 17 '18 at 0:20













          1














          You can skip the np.where, which is redundant here, and just index using np.isin:



          X[np.isin(Y,a)]

          array([[2, 1],
          [1, 3],
          [5, 9],
          [6, 7]])


          This is because np.isin gives you a boolean array of where Y is in a:



          array([False,  True,  True,  True,  True, False])


          So by indexing with this array, it only selects those rows where True






          share|improve this answer





















          • 1





            It works fine, thank you :)

            – Esraa Mohamed
            Nov 17 '18 at 0:20
















          1














          You can skip the np.where, which is redundant here, and just index using np.isin:



          X[np.isin(Y,a)]

          array([[2, 1],
          [1, 3],
          [5, 9],
          [6, 7]])


          This is because np.isin gives you a boolean array of where Y is in a:



          array([False,  True,  True,  True,  True, False])


          So by indexing with this array, it only selects those rows where True






          share|improve this answer





















          • 1





            It works fine, thank you :)

            – Esraa Mohamed
            Nov 17 '18 at 0:20














          1












          1








          1







          You can skip the np.where, which is redundant here, and just index using np.isin:



          X[np.isin(Y,a)]

          array([[2, 1],
          [1, 3],
          [5, 9],
          [6, 7]])


          This is because np.isin gives you a boolean array of where Y is in a:



          array([False,  True,  True,  True,  True, False])


          So by indexing with this array, it only selects those rows where True






          share|improve this answer















          You can skip the np.where, which is redundant here, and just index using np.isin:



          X[np.isin(Y,a)]

          array([[2, 1],
          [1, 3],
          [5, 9],
          [6, 7]])


          This is because np.isin gives you a boolean array of where Y is in a:



          array([False,  True,  True,  True,  True, False])


          So by indexing with this array, it only selects those rows where True







          share|improve this answer














          share|improve this answer



          share|improve this answer








          edited Nov 17 '18 at 0:20

























          answered Nov 17 '18 at 0:16









          saculsacul

          30k41740




          30k41740








          • 1





            It works fine, thank you :)

            – Esraa Mohamed
            Nov 17 '18 at 0:20














          • 1





            It works fine, thank you :)

            – Esraa Mohamed
            Nov 17 '18 at 0:20








          1




          1





          It works fine, thank you :)

          – Esraa Mohamed
          Nov 17 '18 at 0:20





          It works fine, thank you :)

          – Esraa Mohamed
          Nov 17 '18 at 0:20


















          draft saved

          draft discarded




















































          Thanks for contributing an answer to Stack Overflow!


          • Please be sure to answer the question. Provide details and share your research!

          But avoid



          • Asking for help, clarification, or responding to other answers.

          • Making statements based on opinion; back them up with references or personal experience.


          To learn more, see our tips on writing great answers.




          draft saved


          draft discarded














          StackExchange.ready(
          function () {
          StackExchange.openid.initPostLogin('.new-post-login', 'https%3a%2f%2fstackoverflow.com%2fquestions%2f53346940%2fusing-np-where-to-find-element-from-sub-arrays%23new-answer', 'question_page');
          }
          );

          Post as a guest















          Required, but never shown





















































          Required, but never shown














          Required, but never shown












          Required, but never shown







          Required, but never shown

































          Required, but never shown














          Required, but never shown












          Required, but never shown







          Required, but never shown







          Popular posts from this blog

          Guess what letter conforming each word

          Port of Spain

          Run scheduled task as local user group (not BUILTIN)