python画星空的程序_Python热力学图——《星空密码》,python,力图,代码

描述

使用python，生成一个由随机数据构成的热力图

由于我使用的是三通道，因此热力值小的部分是黑色，热力值高的部分是白色

我：看我生成的热力图

女朋友：你生成的是星空图

代码

#encoding: utf-8

from pyecharts.charts import Geo

from pyecharts import options

from pyecharts.globals import GeoType

import pandas as pd

import webbrowser

import matplotlib.pyplot as plt

import numpy as np

import cv2

from PIL import Image

import random

import cmath

gap_size = 400

image_cols = 1148 + gap_size # 横坐标

image_rows = 1501 + gap_size # 纵坐标

image = Image.new("RGB", (image_cols, image_rows))

star_num = 100 # 星星个数

for k in range(star_num):

star_row = random.randint(gap_size/2, image_rows-gap_size/2)

star_col = random.randint(gap_size/2, image_cols-gap_size/2)

impact_cols = [star_col+i-50 for i in range(100)]

impact_rows = [star_row+i-50 for i in range(100)]

# 每个star的影响范围

for wide in impact_cols:

for height in impact_rows:

position_near = (wide, height) # 取全图中的一个点

pixBefore = image.getpixel(position_near) # 取一下当前的像素值

base_value = 200 # 影响基值，现在先写成200

# 影响系数 = 一个和距离有关的值, 或者是其他的

distance = (cmath.sqrt( pow(wide-star_col,2) + pow(height-star_row,2)) ).real

impact_factor = 1 / (distance+1)

# 影响因子大于一定的值再计算

if impact_factor > 0.02:

# 影响的值 = 影响系数 * 影响基值

impact_value = impact_factor * base_value

pixAfter = (pixBefore[0] + int(impact_value), pixBefore[1]+ int(impact_value), pixBefore[2]+ int(impact_value))

image.putpixel(position_near, pixAfter)

image.save("star.png")