注意:该书中后续几张剩余答案及练习题均在数据科学一栏对应的文章中
“大润发”、“沃尔玛”、“联华”和“农工商”四个超市都卖苹果、香蕉、桔子、猕猴桃和芒果5种水果。使用NumPy的ndarray实现以下功能。
a) 创建2个一维数组分别存储超市名称和水果名称;
b) 创建1个4×5的二维数组存储不同超市的水果价格, 的二维数组存储不同超市的水果价格,其中价格由4到10范围内的随机数生成;
c) 选择“大润发”的苹果和“联华”的香蕉,并将价格增加1元;
d) “农工商”水果大减价,所有水果价格减少2元
e)统计四个超市苹果和芒果的销售均价
f)找出桔子价格最贵的超市名称(不是序号)
#创建2个一维数组分别存储超市名称和水果名称;
import numpy as np
supmarkets=np.array(['大润发', '沃尔玛', '联华', '农工商'])
fruits=np.array(['苹果', '香蕉', '桔子', '猕猴桃', '芒果'])
#创建1个4×5的二维数组存储不同超市的水果价格, 的二维数组存储不同超市的水果价格,其中价格由4到10范围内的随机数生成;
price=np.random.randint(4,11,(4,5))
#选择“大润发”的苹果和“联华”的香蕉,并将价格增加1元
price[(supmarkets=='大润发')|(supmarkets=='联华'),(fruits=='苹果')|(fruits=='香蕉')]+1
#“农工商”水果大减价,所有水果价格减少2元
price[supmarkets=='农工商']-2
#统计四个超市苹果和芒果的销售均价
price[:,(fruits=='苹果') | (fruits=='芒果')].mean(axis=0)
#找出桔子价格最贵的超市名称(不是序号)
#argmax--求最大值的索引(注意:求出的索引为行索引)
supmarkets[price[:,fruits=='桔子'].argmax()]