叮！你要的 Python 脱单攻略已送达

最新推荐文章于 2024-01-24 16:37:49 发布

Python开发者

最新推荐文章于 2024-01-24 16:37:49 发布

阅读量408

点赞数

年关将至，学妹却因为男朋友不愿意带她回家过年而难过，我劝学妹主动点，自己多提几次，但是学妹觉得身为女孩子不能太主动。

于是，作为妇女之友的我打算用一个故事来开导学妹，我准备用Python作分析，拿数据说话：

“学妹呀，我来给你讲个故事~”

有一座城市，当地风俗是，想结婚的男子必须先向心仪的女子求婚，而女子则需要等待求婚。

牧师每年会邀请人数相同的适婚男女参与一次集体相亲。一次相亲活动可能有很多轮：

男子会首先向自己最爱的女子求婚，女子则会在所有的追求者中选择她的最爱；
如果男子被拒绝，下一轮会向他第二喜欢的女子求婚；
上一轮已经订婚的女子如果得到她更爱的人的求婚，则会毫不留情地抛弃未婚夫，和更爱的人在一起；
被抛弃的男子需要重新参与求婚。

如此反复，等大家都订婚，就举办集体婚礼。

聪明学妹马上抓住了重点：只要这个活动没有结束，女生就有机会选择她更爱的人！这对女生来说实在是太棒了，只要等着就会等到最合适的那个人！

我点点头：是的没错，但每个人对自己最终伴侣的满意度才是我们更应该关注的。虽然无法实地调查，但作为熟知Python语言和蒙特卡洛模拟方法的数据分析师，透过规律看结果还是非常简单的。看到结果你会明白一些道理。

640?wx_fmt=jpeg

蒙特卡罗方法：一种基于随机试验和统计计算的数值方法。

接下去，就用Python来分析这类婚配规则的最终结果。

我要用Python随机生成数量相等的男女，每个人都有自己的代号和对每一位异性的喜爱程度排名，再用计算机模拟求婚过程，最后看结果。多轮模拟的结果就会更接近真实值。

先生成男性群体——

man = pd.DataFrame( [['w'+str(i) for i in random.sample(range(1,women_num+1),women_num)] 
                      for i in range(man_num)],
                    index = ['m'+str(i) for i in range(1,man_num+1)],
                    columns = ['level'+str(i) for i in range(1,women_num+1)]
                    )

人数设定为100人，生成的结果是这样的，包含每个人的代号，和对异性群体中每个人的喜好程度：

640?wx_fmt=gif

每天上午，每位还没订婚的男子，会向还没拒绝过他的女子中，他最爱的那一个求婚；
每天下午，每位女性会在自己接到的求婚信中，选择她最中意的男子接受求婚；如果没接到求婚，就继续等待。

这个过程实现起来可能要等很多天，但我们有Python就方便许多~部分代码长这样：

print('======测试集{}模拟开始====='.format(i))
print('=====开始模拟求婚过程=====')
level_num = 0
while man_ismapping['love_level'].min() == 0:
level_num += 1
print('=====开始第{}天婚姻配对====='.format(level_num))
u_mapping_man = man_ismapping[man_ismapping.target == 'n'].index.tolist()