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 '18 at 3:04









      PythonistaPythonista

      344




      344
























          1 Answer
          1






          active

          oldest

          votes


















          1














          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









            1














            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




























              1














              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


























                1












                1








                1







                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 '18 at 3:11









                Wen-BenWen-Ben

                114k83369




                114k83369
































                    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%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

                    How to change which sound is reproduced for terminal bell?

                    Can I use Tabulator js library in my java Spring + Thymeleaf project?

                    Title Spacing in Bjornstrup Chapter, Removing Chapter Number From Contents