matplotlibで角度と長さの値を入力して作図したい

python
1red = 15
2blue　= 10
3angle = 90

上記のような感じで、値を入力することで下のように作図をしたいです。

下のコードで作図することは何とか出来たのですが、簡単にできないかと思い質問させていただきます。
基礎的なことだろうとは思いますがよろしくお願いします。

pytohon
1plt.plot([0,15],[0,0],c='red')
2plt.plot([15,(15+(10/2**0.5))],[0,(10/2**0.5)],c='blue')
3plt.plot([15,(15+(10/2**0.5))],[0,-(10/2**0.5)],c='blue')

aokikenichi

2020/07/12 21:35

「簡単にできないか」とはどういうことでしょうか？同様の作図を複数するので、逐一書くのが面倒なので関数化したい、というようなことでしょうか。

yonhark0

2020/07/14 07:07

すみません。言葉が足りませんでした。 aokikenichiさんがおっしゃる通り、同様の作図を複数したいと考えていました。回答もしてくださりありがとうございます！勉強になります。

aokikenichi

2020/07/14 09:14

解決されたようで良かったです

行動規範の内容に同意します

回答1件

ベストアンサー

追記・修正依頼させていただいた想定でよろしければこんなかんじかと

Python
1import matplotlib.pyplot as plt
2import math
3
4def plot_angle(red, blue, angle):
5  plt.plot([0,red],[0,0],c='red')
6  plt.plot([red,(red+blue*math.cos(angle/2/180*math.pi))],[0,blue*math.sin(angle/2/180*math.pi)],c='blue')
7  plt.plot([red,(red+blue*math.cos(angle/2/180*math.pi))],[0,-blue*math.sin(angle/2/180*math.pi)],c='blue')
8
9plot_angle(15, 10, 90)