จัดเรียง 2 รายการใน Python ตามอัตราส่วนขององค์ประกอบที่เกี่ยวข้องแต่ละรายการหรือตามรายการที่สาม

ในหนึ่งบรรทัด:

values, weights = zip(*sorted(zip(values, weights), key=lambda t: t[0]/t[1]))

เพื่ออธิบาย: ขั้นแรก ให้บีบอัดรายการเพื่อจับคู่

pairs = zip(values, weights) 
# [(60, 20), (100, 50), (120, 30)]

จากนั้นจัดเรียงตามผลหารของค่าต่อน้ำหนัก

sorted_pairs = sorted(pairs, key=lambda t: t[0]/t[1]) 
# [(100, 50), (60, 20), (120, 30)]

สุดท้ายให้คลายซิปกลับเข้าไปในรายการแยกกัน

values, weights = zip(*sorted_pairs)
# (100, 60, 120), (50, 20, 30)

อีกทางเลือกหนึ่งคือการสร้างสิ่งอันดับที่มีอัตราส่วนเป็นองค์ประกอบแรกอย่างชัดเจน

ratios, values, weights = zip(*sorted((v/w, v, w) for v, w in zip(values, weights)))

แบบแรกดูเหมือนจะเร็วกว่าเล็กน้อยในการทดสอบอย่างรวดเร็ว หากคุณกำลังมองหาอัลกอริธึมที่เหมาะสมที่สุด คุณอาจจะต้องคลี่คลายสิ่งต่าง ๆ และวิธีแก้ปัญหาจะไม่กระชับ

และเพื่อตอบความคิดเห็นจาก @TomWyllie หากคุณมีรายการอัตราส่วนอยู่แล้ว คุณสามารถใช้:

ratios, values, weights = zip(*sorted(zip(ratios, values, weights)))

โปรดทราบว่าคำตอบสองข้อสุดท้ายนี้แตกต่างจากวิธีแก้ปัญหาเริ่มต้นในกรณีที่สองคู่มีอัตราส่วนเท่ากัน โซลูชันเหล่านี้จะเรียงลำดับตามค่าตามลำดับ ในขณะที่โซลูชันแรกจะเก็บรายการไว้ในลำดับเดียวกันกับรายการเดิม

วิธีที่ยอดเยี่ยมในการทำเช่นนี้คือการสร้างรายการหลายมิติที่มีค่าและน้ำหนัก:

for i in range(len(values)):
    values_and_weights.append([values[i], weights[i])
# The resulting list is [[60, 20], [100, 50], [120, 30]]

จากนั้นใช้วิธีเรียงลำดับโดยมีค่าหารด้วยน้ำหนักเป็นคีย์

values_and_weights.sort(key=(lambda x: x[0]/x[1]))

สำหรับวิธีแก้ปัญหาที่ชัดเจนยิ่งขึ้น (แต่อาจมี pythonic น้อยกว่า) ให้สร้างรายการดัชนี จัดเรียงตามค่าที่ดัชนีนั้นใน value_per_unit และเรียงลำดับ values และ weights ใหม่ตามลำดับ

sorted_indices = [index for index, value in 
                  sorted(enumerate(value_per_unit), key=lambda x: x[1])]
values = [values[i] for i in sorted_indices]
weights = [weights[i] for i in sorted_indices]

print(values, weights)

เอาท์พุต:

([100, 60, 120], [50, 20, 30])

คุณสามารถจัดระเบียบสิ่งนี้ได้ โดยกำจัดการวนซ้ำเพิ่มเติมที่ไม่จำเป็นโดยใช้ zip และด้วยนิพจน์ตัวสร้าง

values, weights = zip(*((values[i], weights[i]) for i, value in
                  sorted(enumerate(value_per_unit), key=lambda x: x[1])))
print(values)
print(weights)

ซึ่งเอาท์พุต;

(100, 60, 120)
(50, 20, 30)

โปรดทราบว่าค่าสุดท้ายเหล่านี้คือ tuples ไม่ใช่ lists หากคุณต้องการให้ผลลัพธ์เป็นรายการจริงๆ การใช้ values, weights = map(list, (values, weights)) แบบธรรมดาก็เพียงพอแล้ว คุณสามารถรวมเรื่องนั้นไว้ในบรรทัดเดียวได้ แม้ว่าเมื่อถึงจุดนั้น คงจะยากที่จะติดตามสิ่งที่เกิดขึ้น

ปัญหาที่คุณพบเกิดจากการใช้ฟิลด์จากการคำนวณเหนือแต่ละองค์ประกอบ (องค์ประกอบ I จะมีค่าจากการคำนวณ values[I]/weights[I]) เพื่อแก้ปัญหานี้โดยยังคงรักษาให้เข้าใจง่ายมาก คุณสามารถเปลี่ยนให้เป็นทูเพิลของแบบฟอร์มนี้ได้: ( calculated_value, (value, weight) ) ต่ออิลิเมนต์

แนวทางนี้ช่วยให้อ่านและทำความเข้าใจได้ง่าย ดูวิธีแก้ปัญหาต่อไปนี้:

values = [60, 100, 120]
weights = [20, 50, 30]
value_per_unit = []

for I in range(len(values)):
    value_per_unit.append( (values[I]/weights[I], (values[I], weights[I])) )
sorted_value_per_unit = sorted(value_per_unit, key=lambda x: x[0])

sorted_values = []
sorted_weights = []
for I in range(len(values)):
    (value, weight) = sorted_value_per_unit[I][1]
    sorted_values.append(value)
    sorted_weights.append(weight)

print(str(sorted_values))
print(str(sorted_weights))

โปรดทราบว่าฉันได้แก้ไขการวนซ้ำจากโค้ดต้นฉบับของคุณ:

range(values) เปลี่ยนเป็น range(len(values))

เนื่องจาก range ต้องการความยาวของรายการ ไม่ใช่ตัวรายการเอง