Python/生成数据

matplotlib

安装

Linux下安装

sudo apt-get install python3.x-dev python3.x-tk tk-dev
sudo apt-get install libfreetype6-dev g++
pip install --user matplotlib

window下安装
下载matplotlib安装程序。为此，请访问https://pypi.python.org/pypi/matplotlib/，并查找与你使用的Python版本匹配的wheel文件（扩展名为.whl的文件）,例如matplotlib-1.4.3-cp35-none-win32.whl，然后cmd中进入该文件目录下

python -m pip install --user matplotlib-1.4.3-cp35-none-win32.whl

---

plot()绘制直方图

#mpl_squares.py
import matplotlib.pyplot as plt

input_value = [1, 2, 3, 4, 5]
squares = [1, 4, 9, 16, 25]
plt.plot(input_value, squares, linewidth=5)

plt.title("Square Numbers", fontsize=14)
# 设置标题
plt.xlabel("Value", fontsize=14)
plt.ylabel("Square of Value", fontsize=14)
# 设置坐标轴
plt.tick_params(axis="both", labelsize=14)
# 设置坐标刻度
plt.show()

scatter()绘制散点图

# scatter_square.py

import matplotlib.pyplot as plt
from matplotlib.transforms import Bbox

x_values = list(range(1, 1001))
y_values = [x**2 for x in x_values]
plt.scatter(x_values, y_values, c=y_values, cmap=plt.cm.Blues, edgecolor='none', s=20)
# c设置为y值列表，并使用参数cmap定义pyplot使用的颜色映射为蓝色，y值较小显示为浅蓝色，y值较大是显示为深蓝色
plt.title("Square Number", fontsize=24)
plt.xlabel("Value", fontsize=14)
plt.ylabel("Square of Value", fontsize=14)

plt.tick_params(axis="both", which="major", labelsize=14)

plt.axis([0, 1100, 0, 1100000])
# 设置坐标轴范围

#plt.savefig('images\square_plot_png',bbox_inches='tight')
# 用于保存生成的图片
plt.show()

生成随机散点图

建立random_walk<project>，在文件夹下创建random_walk.py, rw_visual.py

# random_walk.py
# 定义RandomWalk对象，用于生成随机散点二维坐标
from random import choice
from re import X

class RandomWalk():
    def __init__(self, num_points=50000):
        self.num_points = num_points
        # 设置初始位置
        self.x_values = [0]
        self.y_values = [0]

    def fill_walk(self):
        while len(self.x_values) < self.num_points:
            x_direction = choice([1, -1])
            # choice()传入一个list，返回list里随机一个值
            x_distance = choice([0, 1, 2, 3, 4])
            x_step = x_direction * x_distance
            y_direction = choice([1, -1])
            y_distance = choice([0, 1, 2, 3, 4])
            y_step = y_direction * y_distance
            
            if x_step == 0 and y_step == 0:
            	# 防止停在原地
                continue
            next_x = self.x_values[-1] + x_step
            next_y = self.y_values[-1] + y_step
            self.x_values.append(next_x)
            self.y_values.append(next_y)

# rw_visual.py
# 用于生成散点图
import matplotlib.pyplot as plt
from random_walk import RandomWalk

while True:
    rw = RandomWalk()
    rw.fill_walk()
    # 设置绘图窗口的尺寸
    plt.figure(dpi=128)
    point_numbers = list(range(rw.num_points))
    plt.scatter(
        rw.x_values,
        rw.y_values,
        c=point_numbers,
        cmap=plt.cm.Blues,
        edgecolor="none",
        s=1,
    )
    plt.scatter(0, 0, c="green", edgecolors="none", s=50)
    plt.scatter(rw.x_values[-1], rw.y_values[-1], c="red", edgecolors="none", s=50)
    # 隐藏axis
    # plt.axis("off")
    plt.show()

    keep_running = input("make another walk?(y/n):")
    if keep_running == "n":
        break

---

pygal

使用Python可视化包pygal来生成可缩放的矢量图文件；在此项目将生成一个表示掷骰子结果的数据集，并根据结果绘制出一个图形

安装

Linux系统下安装

pip install --user pygal==x.x #对应的版本号

Windows系统下安装

python -m pip install --user pygal==x,x

了解使用pygal可创建什么样什么样的图表，可以查看图表类型画廊
pygal图表类型画廊

绘制模拟掷骰子点数直方图

# die.py
# 创建一个Die类，生成骰子对象

from random import randint

class Die():
    def __init__(self, num_sides=6):
        self.num_sides=num_sides
    def roll(self):
        #返回一个1与num_sides之间的随机值
        return randint(1,self.num_sides)

模拟掷骰子，并且将结果统计出来；有了频率列表后，使用pygal.Bar()绘制一个表示结果的直方图

# die_visual.py
# 模拟投掷一个骰子
import pygal
from die import Die

die=Die()
results=[]
for roll_num in range(100):
    result=die.roll()
    results.append(result)

frequencies=[]
for value in range(1, die.num_sides+1):
    #统计results里1，2，3，4，5，6出现的次数且存进frequencies里
    frequency=results.count(value)
    frequencies.append(frequency)
print(frequencies)

# 对所获得的frequencies进行可视化操作
hist = pygal.Bar()
hist._title = 'Results of rolling one D6 1000 times'
hist.x_labels = ['1', '2', '3', '4', '5', '6']
# 设置x轴坐标数值
hist._x_title = 'Result'
hist._y_title = 'Frequency of Result'
hist.add('D6', frequencies)
hist.render_to_file('images/die_visual.svg')
# 保存成svg文件输出

---

同时投掷两个骰子

# die_visual.py
# 模拟投掷两个骰子
import pygal
from die import Die

die_1=Die()
die_2=Die()
results=[]
for roll_num in range(100):
    result=die_1.roll()+die_2.roll()
    results.append(result)

frequencies=[]
for value in range(2, die_1.num_sides+die_2.num_sides+1):
    #统计results里1，2，3，4，5，6出现的次数且存进frequencies里
    frequency=results.count(value)
    frequencies.append(frequency)
print(frequencies)

# 对所获得的frequencies进行可视化操作
hist = pygal.Bar()
hist._title = 'Results of rolling two D6 100 times'
hist.x_labels = ['1', '2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12']
# 设置x轴坐标数值
hist._x_title = 'Result'
hist._y_title = 'Frequency of Result'
hist.add('D6+D6', frequencies)
# 使用add()方法将一系列值添加到图表中；向它传递要给添加的值指定的标签，还有一个列表，其中包含将出现在图表中的值
hist.render_to_file('images/die_visual.svg')
# 保存成svg文件输出