numpy基础属性方法随机整理（七）--- np.take() / np.where / np.argmax()/np.argmin()（对照pandas相应功能的属性和方法对比学习）

Note1:
np.take(close_prices, np.where(weekdays==weekday))
pandas中参考 《pandas基础属性方法随机整理（四）—实例梳理（多知识点）》有三种方法实现该条件查询功能：

1. 方法1：辅助列 data[‘Mon’]， data[data[‘Mon’]==1].mean() # 掩码

   即 数组[关系表达式]：
   关系表达式是一个布尔型书序，其中为True的元素对应于数组中满足关系表达式的元素，以上下标运算的值就是从数组中挑选与布尔数组中为True的元素相对应的元素

2. 方法2：辅助列data[‘Mon’]， # 查询.query, 功能与索引一样，有时更方便
3. 方法3： 不添加辅助列，利用groupby()方法实现降采样 data.groupby ( lambda x: x.month ). mean(). T[1]

numpy方法：
for循环中的条件查询np.take(.., np.where(..))
np.where(关系表达式)：数组中满足关系表达式的元素的下标数组
np.take(数组，下标数组)：数组中由下标数组所表示的元素集合

where(condition, [x, y])：
    -- x, y不为空: condition为True, 返回x; False, 返回y
    -- x, y为空： 返回condition为True的数组下标

average_prices = np.zeros(5)
for weekday in range(average_prices.size):
    average_prices[weekday] = np.take(close_prices,
    np.where(weekdays==weekday)).mean()
    # np.where(weekdays==weekday)获得序列号的数组，作为take取元素的下标

import numpy as np
c = ages + scores * 1j                              # j要用 1j 表示
d = np.sort_complex(c).real                         # float:[20. 22. 23. 27. 29.] <class 'numpy.ndarray'>
d = d.astype('int64')                               # int:[20 22 23 27 29] <class 'numpy.ndarray'>

index_names = []
for i, v in enumerate(d):
    print(np.where(ages ==v)[0][0], np.take(ages, np.where(ages == v))[0][0])
    index_names.append(np.where(ages ==v)[0][0])    # 安装d的元素顺序依次对比，取出ages的下标

namesComplexSorted = names[index_names]
print(namesComplexSorted)                           # ['Jue' 'Kity' 'Mike' 'scote' 'Tom']
print(np.where(namesComplexSorted == 'Kity')[0][0]) # (array([1], dtype=int64),)

Note2:
最大值 / 最小值对应的索引：

属性描述	numpy	pandas
最大值的索引	argmax()	idxmax()
最小值的索引	argmin()	idxmin()

Note3:
print()内部的if…else…语句：

print(average_prices,
        '(max)' if(weekday == max_index) else
        '(min)' if(weekday == min_index) else '')

---

Code:

import os 
import sys
import datetime as dt 
import numpy as np 


g_weekdays = ('MON','Tue','WED','THU','FRI','SAT','SUN')

def dmy2weekday(dmy):
    return dt.datetime.strptime(str(dmy, encoding='utf-8'), '%d-%m-%Y').date().weekday()


def read_data(filename):
    weekdays, close_prices = np.loadtxt(filename, delimiter=',', usecols=(1,6),unpack=True,converters={1:dmy2weekday})
    return weekdays, close_prices

def cal_average_prices(weekdays, close_prices):
    average_prices = np.zeros(5)
    for weekday in range(average_prices.size):
        average_prices[weekday] = np.take(
            close_prices, np.where(weekdays==weekday)).mean()
        # np.take(close_prices, np.where().mean()) 类比pandas中的 df[df[weekday==1]].mean()
        # 参考 《pandas基础属性方法随机整理（四）---实例梳理（多知识点）》 
    return average_prices

def main(argc, argv, envp):
    weekdays, close_prices = read_data('aapl.csv')
    average_prices = cal_average_prices(weekdays, close_prices)
    max_index = np.argmax(average_prices)
    # pandas中对应的属性为 idxmax/ idxmin
    min_index = np.argmin(average_prices)
    for weekday, average_prices in enumerate(average_prices):
        print(g_weekdays[weekday], ':', average_prices,
            # weekday 作为g_weekday的索引号
            '(max)' if(weekday == max_index) else
            '(min)' if(weekday == min_index) else '')
        # print() print内部竟然还有这种if..else..操作
    return 0

if __name__ == '__main__':
    sys.exit(main(len(sys.argv), sys.argv, os.environ))