同じ数式を２度使わないで処理

Question

イメージ説明

前提・実現したいこと

if文を使った関数を1つにし、変数を処理し、グラフを2つ表示したいです。

結果としては添付の画像の様にしたいです。

該当のソースコード

import matplotlib.pyplot as plt

a = 1200
b = 1800
L = 3000

print("E")
E = input()
E = int(E)

print("I")
I = input()
I = int(I)

print("P")
P = input()
P = int(P)

print("E2")
E2 = input()
E2 = int(E2)

print("I2")
I2 = input()
I2 = int(I2)

print("P2")
P2=input()
P2=int(P2)

def f(x_i) :
  if x_i < a:
    return (b * P) * (-x_i**3 + a * (a + 2 * b) * x_i) / (6 * E * I * L)
  else:
    return (a * P) / (6 * E * I * L) * (-(L - x_i)**3 + b * (2 * a + b) * (L - x_i))

def k(x_i) :
  if x_i < a:
    return (b * P2) * (-x_i**3 + a * (a + 2 * b) * x_i) / (6 * E2 * I2 * L)
  else:
    return (a * P2) / (6 * E2 * I2 * L) * (-(L - x_i)**3 + b * (2 * a + b) * (L - x_i))

x = list(range(L))
y=list(map(f,x))
y2=list(map(k,x))

plt.grid(which="major", color="gray", linestyle="-")
plt.xlim(0, L)
plt.ylim(min(y), max(y) + 0.5)



plt.plot(x, y, label = "label-A")
plt.plot(x,y2, label = "label-B")
plt.xlabel("x[mm]")
plt.ylabel("y[mm]")
plt.legend(loc = 'lower right')
plt.show()

Answer 1

引数にx_iとP , E , Iも受け取るようにし、受け取ったものを利用するようにすれば解決できると思います

Comments

[shion_shion]

すみません、プログラミングがほぼ初心者で分からない事が多く
実際にその部分だけでいいのでコードなど見せて頂けるとありがたいです...

[norapomu]

def f(x_i,P,E,I) :
if x_i < a:
return (b * P) * (-x_i**3 + a * (a + 2 * b) * x_i) / (6 * E * I * L)
else:
return (a * P) / (6 * E * I * L) * (-(L - x_i)**3 + b * (2 * a + b) * (L - x_i))
例ですが、上記のように変更し、呼ぶ際に引数を変更することで計算結果が変わるかと思います。
動作に関しては、どういう動作をすれば正しいのかも不明なので未検証ですが。

[shion_shion]

動作的にはif文を２回使わずに追加で載せた画像のような結果を得たいと思っています。

[norapomu]

まず、関数宣言の行を下記のように変更し
def f(x_i) :　→　def f(x_i,E,I,P) :
関数を呼び出す行を下記のように変更してください。
y=list(map(f,x))　→　y=[f(x,E,I,P) for x in range(L)]
画像のような表が出たことは確認済みですが、この結果で正しいかどうかはわからないので確認してください。

[shion_shion]

すみません、説明が足りなかったのですが、for文ではなくmapを使ってという条件があるのですが、mapのままで可能でしょうか？？

[norapomu]

本来は別件の質問になるので、別な質問で募集した方がいいかと思います。

関数をこのように変更してください。
引数に指定していたものをただ配列にしただけです。
def f(x_i,li) :
if x_i < a:
return (b * li[2]) * (-x_i**3 + a * (a + 2 * b) * x_i) / (6 * li[0] * li[1] * L)
else:
return (a * li[2]) / (6 * li[0] * li[1] * L) * (-(L - x_i)**3 + b * (2 * a + b) * (L - x_i))

関数を呼び出す際の処理が大きく変わります。
li = [[E,I,P] for i in range(L)]
arg=map(f,x,li)
y=list(arg)

とすることで、mapを利用しての出力が出来ます。

詳しく知りたい場合は「python map関数 複数引数」のようなキーワードで検索して調べてください。

[actorbug]

単純にlambdaを使うのではだめなのでしょうか。
def f(x_i) :　→　def f(x_i,E,I,P) :
y=list(map(f,x))　→　y=list(map(lambda x_i:f(x_i,E,I,P),x))

[norapomu]

actorbugさん
lambda式を自分が使ったことがなかったため、上のような回答をしてました。
勉強になります、ありがとうございます。

[shion_shion]

何度もお聞きし、ご迷惑をおかけして、すみませんでした。
おかげで解決する事が出来ました。ありがとうございました。
actorbugさんもありがとうございました。

[actorbug]

こちらこそ、横からすみませんでした。