Create a new column based on the values of the column by ID












1














Please help me solve this problem. I searched online but couldn’t find the solution.



df

Id Fruit
1 Apple
1 Carrot
2 Apple
3 Carrot

list = [fruit, vegetable, both]


The question is, how to assign for each id, if id’s fruit column contains apple or carrot the list. If id contains apple then assign ID 1 fruit, if carrot only, then vegetable, if both then both.



I tried to use str.contains by row, but it doesn’t take ID column.



Result



Newdf

Id Fruit New_col
1 Apple Both
1 Carrot Both
2 Apple Fruit
3 Carrot Vegetable


Thank you a lot.










share|improve this question



























    1














    Please help me solve this problem. I searched online but couldn’t find the solution.



    df

    Id Fruit
    1 Apple
    1 Carrot
    2 Apple
    3 Carrot

    list = [fruit, vegetable, both]


    The question is, how to assign for each id, if id’s fruit column contains apple or carrot the list. If id contains apple then assign ID 1 fruit, if carrot only, then vegetable, if both then both.



    I tried to use str.contains by row, but it doesn’t take ID column.



    Result



    Newdf

    Id Fruit New_col
    1 Apple Both
    1 Carrot Both
    2 Apple Fruit
    3 Carrot Vegetable


    Thank you a lot.










    share|improve this question

























      1












      1








      1







      Please help me solve this problem. I searched online but couldn’t find the solution.



      df

      Id Fruit
      1 Apple
      1 Carrot
      2 Apple
      3 Carrot

      list = [fruit, vegetable, both]


      The question is, how to assign for each id, if id’s fruit column contains apple or carrot the list. If id contains apple then assign ID 1 fruit, if carrot only, then vegetable, if both then both.



      I tried to use str.contains by row, but it doesn’t take ID column.



      Result



      Newdf

      Id Fruit New_col
      1 Apple Both
      1 Carrot Both
      2 Apple Fruit
      3 Carrot Vegetable


      Thank you a lot.










      share|improve this question













      Please help me solve this problem. I searched online but couldn’t find the solution.



      df

      Id Fruit
      1 Apple
      1 Carrot
      2 Apple
      3 Carrot

      list = [fruit, vegetable, both]


      The question is, how to assign for each id, if id’s fruit column contains apple or carrot the list. If id contains apple then assign ID 1 fruit, if carrot only, then vegetable, if both then both.



      I tried to use str.contains by row, but it doesn’t take ID column.



      Result



      Newdf

      Id Fruit New_col
      1 Apple Both
      1 Carrot Both
      2 Apple Fruit
      3 Carrot Vegetable


      Thank you a lot.







      python pandas






      share|improve this question













      share|improve this question











      share|improve this question




      share|improve this question










      asked Nov 21 at 3:04









      Pythonista

      224




      224
























          1 Answer
          1






          active

          oldest

          votes


















          0














          Doing map first , then we just using transform nunqiue finding the the position should be 'both'



          s=df.Fruit.map({'Apple':'Fruit','Carrot':'Vegetable'})
          mask=s.groupby(df.Id).transform('nunique').eq(2)
          s[mask]='both'
          df['New']=s
          df
          Out[190]:
          Id Fruit New
          0 1 Apple both
          1 1 Carrot both
          2 2 Apple Fruit
          3 3 Carrot Vegetable





          share|improve this answer





















            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%2f53404704%2fcreate-a-new-column-based-on-the-values-of-the-column-by-id%23new-answer', 'question_page');
            }
            );

            Post as a guest















            Required, but never shown

























            1 Answer
            1






            active

            oldest

            votes








            1 Answer
            1






            active

            oldest

            votes









            active

            oldest

            votes






            active

            oldest

            votes









            0














            Doing map first , then we just using transform nunqiue finding the the position should be 'both'



            s=df.Fruit.map({'Apple':'Fruit','Carrot':'Vegetable'})
            mask=s.groupby(df.Id).transform('nunique').eq(2)
            s[mask]='both'
            df['New']=s
            df
            Out[190]:
            Id Fruit New
            0 1 Apple both
            1 1 Carrot both
            2 2 Apple Fruit
            3 3 Carrot Vegetable





            share|improve this answer


























              0














              Doing map first , then we just using transform nunqiue finding the the position should be 'both'



              s=df.Fruit.map({'Apple':'Fruit','Carrot':'Vegetable'})
              mask=s.groupby(df.Id).transform('nunique').eq(2)
              s[mask]='both'
              df['New']=s
              df
              Out[190]:
              Id Fruit New
              0 1 Apple both
              1 1 Carrot both
              2 2 Apple Fruit
              3 3 Carrot Vegetable





              share|improve this answer
























                0












                0








                0






                Doing map first , then we just using transform nunqiue finding the the position should be 'both'



                s=df.Fruit.map({'Apple':'Fruit','Carrot':'Vegetable'})
                mask=s.groupby(df.Id).transform('nunique').eq(2)
                s[mask]='both'
                df['New']=s
                df
                Out[190]:
                Id Fruit New
                0 1 Apple both
                1 1 Carrot both
                2 2 Apple Fruit
                3 3 Carrot Vegetable





                share|improve this answer












                Doing map first , then we just using transform nunqiue finding the the position should be 'both'



                s=df.Fruit.map({'Apple':'Fruit','Carrot':'Vegetable'})
                mask=s.groupby(df.Id).transform('nunique').eq(2)
                s[mask]='both'
                df['New']=s
                df
                Out[190]:
                Id Fruit New
                0 1 Apple both
                1 1 Carrot both
                2 2 Apple Fruit
                3 3 Carrot Vegetable






                share|improve this answer












                share|improve this answer



                share|improve this answer










                answered Nov 21 at 3:11









                W-B

                99.9k73163




                99.9k73163






























                    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.





                    Some of your past answers have not been well-received, and you're in danger of being blocked from answering.


                    Please pay close attention to the following guidance:


                    • 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%2f53404704%2fcreate-a-new-column-based-on-the-values-of-the-column-by-id%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

                    404 Error Contact Form 7 ajax form submitting

                    How to know if a Active Directory user can login interactively

                    TypeError: fit_transform() missing 1 required positional argument: 'X'