您好,登錄后才能下訂單哦!
基于 itchat 庫來獲取微信好友頭像并執(zhí)行拼接操作,對(duì)微信上文字化好友列表數(shù)據(jù)進(jìn)行可視化展示。
獲取好友頭像
def save_avatar(folder): """ 保存微信好友頭像 :param folder: 保存的文件夾 """ itchat.auto_login(hotReload=True) users = itchat.get_friends() or [] print('%d friends found.' % len(users)) if not os.path.exists(folder): os.makedirs(folder) index = 1 for i, user in enumerate(users): nickname = user.RemarkName username = user.UserName file_path = os.path.join(folder, '%03d_%s.png' % (i, nickname)) if not os.path.isfile(file_path): # 不重復(fù)下載 avatar = itchat.get_head_img(username) with open(file_path, 'w') as f: f.write(avatar) print('Download %d: %s' % (index, file_path)) index += 1
這里只需要傳入一個(gè)保存頭像的文件夾即可,運(yùn)行 itchat.auto_login(hotReload=True) 后會(huì)彈出微信掃碼界面讓你授權(quán)微信登錄,以便接下來的好友數(shù)據(jù)獲取。
在圖片下載時(shí),我添加了一個(gè)防止重復(fù)下載的判斷,以免多次運(yùn)行時(shí)每次都要重新進(jìn)行頭像的下載。
取出待拼接頭像
def get_image_files(folder, filters=None): """ 取出待拼接頭像 :param folder: 目標(biāo)文件夾 :param filters: 需要過濾的圖片 :return: 頭像路徑 """ filters = filters or [] filenames = [os.path.join(folder, sub) for sub in os.listdir(folder) if sub.endswith('.png') and not filters.__contains__(sub)] return filenames
這里單獨(dú)寫個(gè)方法是為了把過濾的邏輯封裝進(jìn)來,以便于去掉指定的微信好友的頭像(比如純色的頭像在拼接之后的大圖看上去很明顯,非強(qiáng)迫癥可忽略)。
計(jì)算拼接的排列
def calculate_align_way(image_num, force_align=False): """ 計(jì)算圖片排版對(duì)齊方式 :param image_num: 圖片數(shù)量 :return: (rowls, columns) """ actual_value = image_num ** 0.5 suggest_value = int(actual_value) if actual_value == suggest_value or force_align: return suggest_value, suggest_value else: return suggest_value, suggest_value + 1
因?yàn)樾枰雷罱K拼接圖片的行列數(shù),所有這里單獨(dú)定義一個(gè)計(jì)算方法。算法就是直接對(duì)圖片總數(shù)開根號(hào),取出的結(jié)果如果正好是整數(shù),就直接返回該結(jié)果。如果不是整數(shù)(大多數(shù)情況都如此),則根據(jù)參數(shù) force_align 來決定是否強(qiáng)制進(jìn)行正好全部鋪滿的顯示。如果設(shè)為 True ,能強(qiáng)制鋪滿,但會(huì)有部分好友未顯示完全;反之則是相對(duì)的情況。 后面發(fā)現(xiàn)拼接圖片最后一行有很多黑色空位時(shí),只需要更改該參數(shù)為True即可。
拼接
def join_images(image_files, rows, cols, width, height, save_file=None): """ 拼接操作 :param image_files: 待拼接的圖片 :param rows: 行數(shù) :param cols: 列數(shù) :param width: 每張小頭像的寬度 :param height: 每張小頭像的高度 :param save_file: 拼接好圖片的保存路徑 """ canvas = np.ones((height * rows, width * cols, 3), np.uint8) for row in range(rows): for col in range(cols): index = row * cols + col if index >= len(image_files): break file_path = image_files[index] im = Image.open(file_path) im = im.resize((width, height)) im_data = np.array(im) if len(im_data.shape) == 2: im_data = np.expand_dims(im_data, -1) x = col * width y = row * height canvas[y: y + height, x: x + width, :] = im_data image = Image.fromarray(canvas) image.show() if save_file: image.save(save_file)
拼接圖片調(diào)用的是科學(xué)計(jì)算包 numpy 和圖片庫 PIL ,主要就是對(duì) ndarray 進(jìn)行操作。
最終將上面的步驟全部串聯(lián)起來,執(zhí)行如下主函數(shù),便得到上面的拼接圖片。
FOLDER = 'avatars' if __name__ == '__main__': # 保存所有好友頭像 save_avatar(FOLDER) # 取到準(zhǔn)備拼接的頭像 image_files = get_image_files(FOLDER) # 計(jì)算拼接的行列 rows, columns = calculate_align_way(len(image_files), force_align=True) # 執(zhí)行拼接操作 join_images(image_files, rows, columns, 64, 64, 'result.png')
Github源碼
以上就是本文的全部?jī)?nèi)容,希望對(duì)大家的學(xué)習(xí)有所幫助,也希望大家多多支持億速云。
免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng),如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱:is@yisu.com進(jìn)行舉報(bào),并提供相關(guān)證據(jù),一經(jīng)查實(shí),將立刻刪除涉嫌侵權(quán)內(nèi)容。