-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmy_function.py
125 lines (121 loc) · 4.87 KB
/
my_function.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
# -*- coding: utf-8 -*-
"""
Created on Sun Dec 8 13:30:11 2019
@author: Azumi Mamiya and Ryota Koyanagi
"""
import numpy as np
import matplotlib.pyplot as plt
def create_android(dpi):
array = [[0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0],
[0,0,0,0,1,0,0,0,0,0,0,1,0,0,0,0],
[0,0,0,0,0,1,0,0,0,0,1,0,0,0,0,0],
[0,0,0,0,0,0,1,0,0,1,0,0,0,0,0,0],
[0,0,0,0,1,1,1,1,1,1,1,1,0,0,0,0],
[0,0,0,1,1,2,1,1,1,1,2,1,1,0,0,0],
[0,0,0,1,1,1,1,1,1,1,1,1,1,0,0,0],
[0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0],
[1,1,0,1,1,1,1,1,1,1,1,1,1,0,1,1],
[1,1,0,1,1,1,1,1,1,1,1,1,1,0,1,1],
[1,1,0,1,1,1,1,1,1,1,1,1,1,0,1,1],
[1,1,0,1,1,1,1,1,1,1,1,1,1,0,1,1],
[1,1,0,1,1,1,1,1,1,1,1,1,1,0,1,1],
[1,1,0,0,1,1,1,1,1,1,1,1,0,0,1,1],
[0,0,0,0,0,1,1,0,0,1,1,0,0,0,0,0],
[0,0,0,0,0,1,1,0,0,1,1,0,0,0,0,0]]
light_green = [164,202,57]
light_pink=[255, 182, 193]
tmp=0
img_array = np.ones((len(array),len(array[0]), 3),np.uint8)*0
for clist in [light_green, light_pink]:
tmp+=1
for i in range(len(array)):
for j in range(len(array)):
for k in range(3):
if array[i][j]==0:
img_array[i][j][k] = 0
elif array[i][j]==1:
img_array[i][j][k] = clist[k]
else:
img_array[i][j][k] = 255
fig = plt.figure()
plt.plot(), plt.imshow(img_array)
plt.axis("off")
fig.patch.set_alpha(0)
plt.savefig('android{}.png'.format(tmp), dpi = dpi)
def create_apple(dpi):
array = [[0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,2,2,2,0,0],
[0,0,0,0,0,0,0,0,3,3,3,0,0,2,2,2,2,2,0,0],
[0,0,0,0,0,0,0,0,0,3,0,0,2,2,2,2,2,0,0,0],
[0,0,0,0,0,0,0,0,0,3,0,2,2,2,2,2,0,0,0,0],
[0,0,0,1,1,1,1,0,0,3,0,2,2,2,2,1,1,0,0,0],
[0,0,1,1,1,1,1,1,0,3,2,2,2,2,1,1,1,1,0,0],
[0,0,1,1,1,1,1,1,3,3,3,2,1,1,1,1,1,1,0,0],
[0,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,0],
[0,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,0],
[1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1],
[1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1],
[1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1],
[0,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,0],
[0,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,0],
[0,0,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,0,0],
[0,0,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,0,0],
[0,0,0,1,1,1,1,1,1,1,1,1,1,1,1,1,1,0,0,0],
[0,0,0,0,1,1,1,1,1,1,1,1,1,1,1,1,0,0,0,0],
[0,0,0,0,0,1,1,1,1,1,1,1,1,1,1,0,0,0,0,0],
[0,0,0,0,0,0,0,1,1,0,0,1,1,0,0,0,0,0,0,0]]
apple_red = [255,0,0]
apple_kuki = [252,205,165]
light_green = [164,202,57]
img_array = np.ones((len(array),len(array[0]), 3),np.uint8)*0
for i in range(len(array)):
for j in range(len(array)):
for k in range(3):
if array[i][j]==0:
img_array[i][j][k] = 0
elif array[i][j]==1:
img_array[i][j][k] = apple_red[k]
elif array[i][j]==2:
img_array[i][j][k] = light_green[k]
else:
img_array[i][j][k] = apple_kuki[k]
fig=plt.figure()
plt.plot(), plt.imshow(img_array)
plt.axis("off")
fig.patch.set_alpha(0)
plt.savefig('apple.png', dpi = dpi)
def create_tux(dpi):
array = [[0,0,0,0,0,2,2,2,2,2,2,0,0,0,0,0],
[0,0,0,0,2,2,2,2,2,2,2,2,0,0,0,0],
[0,0,0,2,2,1,1,2,2,1,1,2,2,0,0,0],
[0,0,0,2,2,1,0,2,2,0,1,2,2,0,0,0],
[0,0,0,0,2,1,1,3,3,1,1,2,0,0,0,0],
[0,0,0,2,2,2,3,1,1,3,2,2,2,0,0,0],
[0,0,2,2,2,1,1,3,3,1,1,2,2,2,0,0],
[0,0,2,2,1,1,1,1,1,1,1,1,2,2,0,0],
[0,2,2,2,1,1,1,1,1,1,1,1,2,2,2,0],
[0,2,2,1,1,1,1,1,1,1,1,1,1,2,2,0],
[0,2,2,1,1,1,1,1,1,1,1,1,1,2,2,0],
[0,2,2,2,1,1,1,1,1,1,1,1,2,2,2,0],
[0,0,3,2,1,1,1,1,1,1,1,1,2,3,0,0],
[0,3,3,2,1,1,1,1,1,1,1,2,2,3,3,0],
[3,3,3,3,2,2,1,1,1,1,2,2,3,3,3,3],
[0,0,3,3,0,0,2,2,2,2,0,0,3,3,0,0]]
gold = [255,185,0]
pen_rinkaku = [150,150,150]
img_array = np.ones((len(array),len(array[0]), 3),np.uint8)*0
for i in range(len(array)):
for j in range(len(array)):
for k in range(3):
if array[i][j]==0:
img_array[i][j][k] = 0
elif array[i][j]==1:
img_array[i][j][k] = 255
elif array[i][j]==2:
img_array[i][j][k] = pen_rinkaku[k]
else:
img_array[i][j][k] = gold[k]
fig=plt.figure()
plt.plot(), plt.imshow(img_array)
plt.axis("off")
fig.patch.set_alpha(0)
plt.savefig('tux.png', dpi = dpi)