将 python pandas df 中的列生日映射到占星术符号

Question

我有一个包含个人生日的列的数据框。我想使用我找到的代码（如下）将该列映射到个人的占星术符号。我无法编写代码来创建变量。我当前的数据框看起来像这样     birthdate  answer  YEAR    MONTH-DAY    1970-03-31    5    1970    03-31    1970-05-25    9    1970    05-25    1970-06-05    3    1970    06-05    1970-08-28    2    1970    08-28我发现创建日期映射函数的代码可在此网站上找到：https ://www.geeksforgeeks.org/program-display-astrological-sign-zodiac-sign-given-date-birth/任何提示将不胜感激。

Answer 1

Series.dt.month_name用小写字符串更改先前的答案：

def zodiac_sign(day, month): 

    # checks month and date within the valid range 

    # of a specified zodiac 

    if month == 'december': 

        return 'Sagittarius' if (day < 22) else 'capricorn'

    elif month == 'january': 

        return 'Capricorn' if (day < 20) else 'aquarius'

    elif month == 'february': 

        return 'Aquarius' if (day < 19) else 'pisces'

    elif month == 'march': 

        return 'Pisces' if (day < 21) else 'aries'

    elif month == 'april': 

        return 'Aries' if (day < 20) else 'taurus'

    elif month == 'may': 

        return 'Taurus' if (day < 21) else 'gemini'

    elif month == 'june': 

        return 'Gemini' if (day < 21) else 'cancer'

    elif month == 'july': 

        return 'Cancer' if (day < 23) else 'leo'

    elif month == 'august': 

        return 'Leo' if (day < 23) else 'virgo'

    elif month == 'september': 

        return 'Virgo' if (day < 23) else 'libra'

    elif month == 'october': 

        return 'Libra' if (day < 23) else 'scorpio'

    elif month == 'november': 

        return 'scorpio' if (day < 22) else 'sagittarius'

dates =  pd.to_datetime(astrology['birthdate'])

y = dates.dt.year

now = pd.to_datetime('now').year

astrology = astrology.assign(month = dates.dt.month_name().str.lower(),

                             day = dates.dt.day,

                             year = y.mask(y > now, y - 100))

print (astrology)

    birthdate  answer  YEAR MONTH-DAY   month  day  year

0  1970-03-31       5  1970     03-31   march   31  1970

1  1970-05-25       9  1970     05-25     may   25  1970

2  1970-06-05       3  1970     06-05    june    5  1970

3  1970-08-28       2  1970     08-28  august   28  1970

astrology['sign'] = astrology.apply(lambda x: zodiac_sign(x['day'], x['month']), axis=1)

print (astrology)

    birthdate  answer  YEAR MONTH-DAY   month  day  year    sign

0  1970-03-31       5  1970     03-31   march   31  1970   aries

1  1970-05-25       9  1970     05-25     may   25  1970  gemini

2  1970-06-05       3  1970     06-05    june    5  1970  Gemini

3  1970-08-28       2  1970     08-28  august   28  1970   virgo

Answer 2

您可以将zodiac_sign函数应用于数据框 -

import pandas as pd

from io import StringIO

# Sample

x = StringIO("""birthdate,answer,YEAR,MONTH-DAY

1970-03-31,5,1970,03-31

1970-05-25,9,1970,05-25

1970-06-05,3,1970,06-05

1970-08-28,2,1970,08-28

""")

df = pd.read_csv(x, sep=',')

df['birthdate'] = pd.to_datetime(df['birthdate'])

df['zodiac_sign'] = df['birthdate'].apply(lambda x: zodiac_sign(x.day, x.strftime("%B").lower()))

print(df)

输出：

   birthdate  answer  YEAR MONTH-DAY zodiac_sign

0 1970-03-31       5  1970     03-31       aries

1 1970-05-25       9  1970     05-25      gemini

2 1970-06-05       3  1970     06-05      Gemini

3 1970-08-28       2  1970     08-28       virgo