Skip to content

Instantly share code, notes, and snippets.

@dirediredock

dirediredock/step_1.py

Last active June 17, 2022 06:10
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save dirediredock/bf96e3e4cc373077e1d9c620c8f5b5cf to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save dirediredock/bf96e3e4cc373077e1d9c620c8f5b5cf to your computer and use it in GitHub Desktop.
Download ZIP
Building a correlation matrix from scratch in Python
Raw
step_1.py
import matplotlib.pyplot as plt
plt.rcParams.update({"font.sans-serif": "Consolas"})
num_var = 5
fig = plt.figure(figsize=(9, 9))
for i in range(num_var ** 2):
ax = fig.add_subplot(num_var, num_var, i + 1)
ax.text(0.5, 0.5, str(i + 1), fontsize=20, ha="center", va="center")
plt.show()
Raw
step_2.py
import matplotlib.pyplot as plt
plt.rcParams.update({"font.sans-serif": "Consolas"})
num_var = 5
plot_coord = []
for x_coord in range(num_var):
for y_coord in range(num_var):
plot_coord.append([x_coord, y_coord])
fig = plt.figure(figsize=(9, 9))
for i in range(num_var ** 2):
ax = fig.add_subplot(num_var, num_var, i + 1)
ax.get_xaxis().set_visible(False)
ax.get_yaxis().set_visible(False)
ax.text(0.5, 0.5, str(plot_coord[i]), fontsize=15, ha="center", va="center")
plt.show()
Raw
step_3.py
import matplotlib.pyplot as plt
plt.rcParams.update({"font.sans-serif": "Consolas"})
num_var = 5
plot_coord = []
for x_coord in range(num_var):
for y_coord in range(num_var):
plot_coord.append([x_coord, y_coord])
fig = plt.figure(figsize=(9, 9))
for i in range(num_var ** 2):
ax = fig.add_subplot(num_var, num_var, i + 1)
if plot_coord[i][0] < plot_coord[i][1]:
ax.get_xaxis().set_visible(False)
ax.get_yaxis().set_visible(False)
ax.text(0.5, 0.35, str(plot_coord[i]), fontsize=15, ha="center", va="center")
ax.text(0.5, 0.65, "↑", fontsize=25, ha="center", va="center", rotation=-45)
elif plot_coord[i][0] > plot_coord[i][1]:
ax.get_xaxis().set_visible(False)
ax.get_yaxis().set_visible(False)
ax.text(0.5, 0.35, str(plot_coord[i]), fontsize=15, ha="center", va="center")
ax.text(0.5, 0.65, "↑", fontsize=25, ha="center", va="center", rotation=135)
else:
ax.axis("off")
ax.text(
0.5,
0.5,
"Feature\n" + str(plot_coord[i][0] + 1),
fontsize=20,
ha="center",
va="center",
)
plt.show()
Raw
step_4.py
import matplotlib.pyplot as plt
plt.rcParams.update({"font.sans-serif": "Consolas"})
num_var = 15
plot_coord = []
for x_coord in range(num_var):
for y_coord in range(num_var):
plot_coord.append([x_coord, y_coord])
fig = plt.figure(figsize=(9, 9))
for i in range(num_var ** 2):
ax = fig.add_subplot(num_var, num_var, i + 1)
if plot_coord[i][0] < plot_coord[i][1]:
ax.get_xaxis().set_visible(False)
ax.get_yaxis().set_visible(False)
ax.text(0.5, 0.35, str(plot_coord[i]), fontsize=15, ha="center", va="center")
ax.text(0.5, 0.65, "↑", fontsize=25, ha="center", va="center", rotation=-45)
elif plot_coord[i][0] > plot_coord[i][1]:
ax.get_xaxis().set_visible(False)
ax.get_yaxis().set_visible(False)
ax.text(0.5, 0.35, str(plot_coord[i]), fontsize=15, ha="center", va="center")
ax.text(0.5, 0.65, "↑", fontsize=25, ha="center", va="center", rotation=135)
else:
ax.axis("off")
ax.text(
0.5,
0.5,
"Feature\n" + str(plot_coord[i][0] + 1),
fontsize=20,
ha="center",
va="center",
)
plt.show()
Raw
step_5.py
import matplotlib.pyplot as plt
plt.rcParams.update({"font.sans-serif": "Consolas"})
num_var = 15
plot_coord = []
for x_coord in range(num_var):
for y_coord in range(num_var):
plot_coord.append([x_coord, y_coord])
font_size = 90 / num_var
fig = plt.figure(figsize=(9, 9))
for i in range(num_var ** 2):
ax = fig.add_subplot(num_var, num_var, i + 1)
if plot_coord[i][0] < plot_coord[i][1]:
ax.get_xaxis().set_visible(False)
ax.get_yaxis().set_visible(False)
ax.text(
0.5, 0.35, str(plot_coord[i]), fontsize=font_size, ha="center", va="center"
)
ax.text(
0.5,
0.65,
"↑",
fontsize=font_size * 2.2,
ha="center",
va="center",
rotation=-45,
)
elif plot_coord[i][0] > plot_coord[i][1]:
ax.get_xaxis().set_visible(False)
ax.get_yaxis().set_visible(False)
ax.text(
0.5, 0.35, str(plot_coord[i]), fontsize=font_size, ha="center", va="center"
)
ax.text(
0.5,
0.65,
"↑",
fontsize=font_size * 2.2,
ha="center",
va="center",
rotation=135,
)
else:
ax.axis("off")
ax.text(
0.5,
0.5,
"Feature\n" + str(plot_coord[i][0] + 1),
fontsize=font_size * 1.2,
weight="bold",
ha="center",
va="center",
)
plt.show()
Raw
step_6.py
import matplotlib.pyplot as plt
plt.rcParams.update({"font.sans-serif": "Consolas"})
num_var = 15
plot_coord = []
for x_coord in range(num_var):
for y_coord in range(num_var):
plot_coord.append([x_coord, y_coord])
font_size = 90 / num_var
fig = plt.figure(figsize=(9, 9))
for i in range(num_var ** 2):
ax = fig.add_subplot(num_var, num_var, i + 1)
if plot_coord[i][0] < plot_coord[i][1]:
ax.get_xaxis().set_visible(False)
ax.get_yaxis().set_visible(False)
ax.text(
0.5, 0.35, str(plot_coord[i]), fontsize=font_size, ha="center", va="center"
)
ax.text(
0.5,
0.65,
"↑",
fontsize=font_size * 2.2,
ha="center",
va="center",
rotation=-45,
)
elif plot_coord[i][0] > plot_coord[i][1]:
ax.get_xaxis().set_visible(False)
ax.get_yaxis().set_visible(False)
ax.text(
0.5, 0.35, str(plot_coord[i]), fontsize=font_size, ha="center", va="center"
)
ax.text(
0.5,
0.7,
"↑",
fontsize=font_size * 2.2,
ha="center",
va="center",
rotation=135,
)
else:
ax.axis("off")
ax.text(
0.5,
0.5,
"Feature\n" + str(plot_coord[i][0] + 1),
fontsize=font_size * 1.2,
weight="bold",
ha="center",
va="center",
)
margin = 0.015
fig.subplots_adjust(
top=1 - margin,
bottom=0 + margin,
right=1 - margin,
left=0 + margin,
hspace=10 * margin,
wspace=10 * margin,
)
plt.show()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment