1376 lines
51 KiB
Plaintext
1376 lines
51 KiB
Plaintext
{
|
||
"cells": [
|
||
{
|
||
"cell_type": "code",
|
||
"id": "e19ec4ae5347c678",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:04.063825Z",
|
||
"start_time": "2024-09-08T08:06:03.748257Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"import numpy as np\n",
|
||
"import pandas as pd\n",
|
||
"\n",
|
||
"pd.options.display.max_columns = 100"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 1
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "initial_id",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:04.504512Z",
|
||
"start_time": "2024-09-08T08:06:04.064830Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"\n",
|
||
"df_crop_details = pd.read_excel('./data/2.xlsx', sheet_name=1)"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 2
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "1d2a51b3414be94",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:04.520325Z",
|
||
"start_time": "2024-09-08T08:06:04.505719Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 去除空格\n",
|
||
"df_crop_details['cropName'] = df_crop_details['cropName'].apply(lambda x: x.strip())\n",
|
||
"# CropType = [x.strip() for x in CropType]\n"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 3
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "514cd9136d9ca341",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:04.551546Z",
|
||
"start_time": "2024-09-08T08:06:04.522370Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"df_crop_planting = pd.read_excel('./data/2.xlsx', sheet_name=0)\n",
|
||
"# 照例去除一下空格\n",
|
||
"df_crop_planting['cropName'] = df_crop_planting['cropName'].apply(lambda x: x.strip())\n",
|
||
"# ffill\n",
|
||
"df_crop_planting['landName'] = df_crop_planting['landName'].ffill()\n",
|
||
"# df_crop_planting"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 4
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "1503f8b642c842db",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:04.583467Z",
|
||
"start_time": "2024-09-08T08:06:04.553055Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"df_land = pd.read_excel('./data/1.xlsx', sheet_name=0)\n",
|
||
"# 去除landType和landName的空格\n",
|
||
"df_land['landType'] = df_land['landType'].apply(lambda x: x.strip())\n",
|
||
"df_land['landName'] = df_land['landName'].apply(lambda x: x.strip())\n",
|
||
"# df_land"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 5
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "3cdf51a9a9d4d30f",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:04.598934Z",
|
||
"start_time": "2024-09-08T08:06:04.585407Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 计算利润\n",
|
||
"unit_profit_lsc = []\n",
|
||
"for line in df_crop_details.values:\n",
|
||
" s = str(line[7]).split('-')\n",
|
||
" unit_profit_lsc.append((float(s[0]) + float(s[1])) / 2 * line[5] - line[6])\n",
|
||
"df_crop_details['unitProfit'] = unit_profit_lsc\n",
|
||
"# df_crop_details"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 6
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "569016a9b90f841b",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:04.629979Z",
|
||
"start_time": "2024-09-08T08:06:04.599929Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"df_crop_type_land = pd.read_excel('./data/1.xlsx', sheet_name=1)\n",
|
||
"# 老规矩，去掉cropName和cropType的空格\n",
|
||
"df_crop_type_land['cropType'] = df_crop_type_land['cropType'].apply(lambda x: x.strip())\n",
|
||
"df_crop_type_land['cropName'] = df_crop_type_land['cropName'].apply(lambda x: x.strip())\n",
|
||
"# df_crop_type_land"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 7
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "a7661d84217b578c",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:04.646043Z",
|
||
"start_time": "2024-09-08T08:06:04.631981Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 搞一下季节和年份的集合\n",
|
||
"SeasonType = [\"单季\", \"第一季\", \"第二季\"]\n",
|
||
"SeasonDict = {\"单季\": 1, \"第一季\": 1, \"第二季\": 2}\n",
|
||
"SeasonNum = [1, 2]\n",
|
||
"years = [2024, 2025, 2026, 2027, 2028, 2029, 2030]"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 8
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "4a4e06a1f2d4d8ab",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:04.661620Z",
|
||
"start_time": "2024-09-08T08:06:04.647030Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 枚举地块类型\n",
|
||
"LandType = {\"A\": \"平旱地\", \"B\": \"梯田\", \"C\": \"山坡地\", \"D\": \"水浇地\", \"E\": \"普通大棚\", \"F\": \"智慧大棚\"}\n",
|
||
"# LandType"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 9
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "cc938565a63a8129",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:04.677628Z",
|
||
"start_time": "2024-09-08T08:06:04.664102Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 枚举地块\n",
|
||
"LandName = df_crop_planting['landName'].unique()\n",
|
||
"# LandName"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 10
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "dae53a6215cea525",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:04.693684Z",
|
||
"start_time": "2024-09-08T08:06:04.679628Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 枚举地块面积，取df_land的landName为key，landArea为value\n",
|
||
"LandArea = {x: df_land[df_land['landName'] == x]['landArea'].values[0] for x in LandName}\n",
|
||
"# LandArea"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 11
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "7a0e6cc209d93b56",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:04.709206Z",
|
||
"start_time": "2024-09-08T08:06:04.694711Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 读入作物名称\n",
|
||
"CropName = df_crop_details['cropName'].unique()\n",
|
||
"# CropName"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 12
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "8a280f918bb3139a",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:04.740752Z",
|
||
"start_time": "2024-09-08T08:06:04.710206Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 作物分类\n",
|
||
"CropType = {x: df_crop_type_land[df_crop_type_land['cropType'] == x]['cropName'].values for x in\n",
|
||
" df_crop_type_land['cropType'].values}\n",
|
||
"CropType['粮食（除了水稻）'] = CropType['粮食'][:-1] # 这样不太好，但能用\n",
|
||
"CropType['豆类'] = np.array(list(CropType['粮食（豆类）']) + list(CropType['蔬菜（豆类）']), dtype=object)\n",
|
||
"CropType['全粮食'] = np.array(list(CropType['粮食']) + list(CropType['粮食（豆类）']), dtype=object)\n",
|
||
"CropType['全粮食（除了水稻）'] = np.array(list(CropType['粮食'][:-1]) + list(CropType['粮食（豆类）']), dtype=object)\n",
|
||
"CropType['全蔬菜'] = np.array(list(CropType['蔬菜']) + list(CropType['蔬菜（豆类）']), dtype=object)\n",
|
||
"CropType['特殊蔬菜'] = np.array([\"大白菜\", \"白萝卜\", \"红萝卜\"], dtype=object)\n",
|
||
"CropType[\"普通蔬菜\"] = np.array([c for c in CropType['全蔬菜'] if c not in CropType['特殊蔬菜']], dtype=object)\n",
|
||
"CropType"
|
||
],
|
||
"outputs": [
|
||
{
|
||
"data": {
|
||
"text/plain": [
|
||
"{'粮食（豆类）': array(['黄豆', '黑豆', '红豆', '绿豆', '爬豆'], dtype=object),\n",
|
||
" '粮食': array(['小麦', '玉米', '谷子', '高粱', '黍子', '荞麦', '南瓜', '红薯', '莜麦', '大麦', '水稻'],\n",
|
||
" dtype=object),\n",
|
||
" '蔬菜（豆类）': array(['豇豆', '刀豆', '芸豆'], dtype=object),\n",
|
||
" '蔬菜': array(['土豆', '西红柿', '茄子', '菠菜', '青椒', '菜花', '包菜', '油麦菜', '小青菜', '黄瓜',\n",
|
||
" '生菜', '辣椒', '空心菜', '黄心菜', '芹菜', '大白菜', '白萝卜', '红萝卜'], dtype=object),\n",
|
||
" '食用菌': array(['榆黄菇', '香菇', '白灵菇', '羊肚菌'], dtype=object),\n",
|
||
" '粮食（除了水稻）': array(['小麦', '玉米', '谷子', '高粱', '黍子', '荞麦', '南瓜', '红薯', '莜麦', '大麦'],\n",
|
||
" dtype=object),\n",
|
||
" '豆类': array(['黄豆', '黑豆', '红豆', '绿豆', '爬豆', '豇豆', '刀豆', '芸豆'], dtype=object),\n",
|
||
" '全粮食': array(['小麦', '玉米', '谷子', '高粱', '黍子', '荞麦', '南瓜', '红薯', '莜麦', '大麦', '水稻',\n",
|
||
" '黄豆', '黑豆', '红豆', '绿豆', '爬豆'], dtype=object),\n",
|
||
" '全粮食（除了水稻）': array(['小麦', '玉米', '谷子', '高粱', '黍子', '荞麦', '南瓜', '红薯', '莜麦', '大麦', '黄豆',\n",
|
||
" '黑豆', '红豆', '绿豆', '爬豆'], dtype=object),\n",
|
||
" '全蔬菜': array(['土豆', '西红柿', '茄子', '菠菜', '青椒', '菜花', '包菜', '油麦菜', '小青菜', '黄瓜',\n",
|
||
" '生菜', '辣椒', '空心菜', '黄心菜', '芹菜', '大白菜', '白萝卜', '红萝卜', '豇豆', '刀豆',\n",
|
||
" '芸豆'], dtype=object),\n",
|
||
" '特殊蔬菜': array(['大白菜', '白萝卜', '红萝卜'], dtype=object),\n",
|
||
" '普通蔬菜': array(['土豆', '西红柿', '茄子', '菠菜', '青椒', '菜花', '包菜', '油麦菜', '小青菜', '黄瓜',\n",
|
||
" '生菜', '辣椒', '空心菜', '黄心菜', '芹菜', '豇豆', '刀豆', '芸豆'], dtype=object)}"
|
||
]
|
||
},
|
||
"execution_count": 13,
|
||
"metadata": {},
|
||
"output_type": "execute_result"
|
||
}
|
||
],
|
||
"execution_count": 13
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "bf6d61d66f06bdeb",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:04.756788Z",
|
||
"start_time": "2024-09-08T08:06:04.741752Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# any([x[-1] == ' ' for x in CropName])\n",
|
||
"# 居然有空格\n",
|
||
"# 现在去掉了"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 14
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "22f9730e3d9c1016",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:04.945877Z",
|
||
"start_time": "2024-09-08T08:06:04.757790Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 亩产量\n",
|
||
"unit_yield_lsc = {\n",
|
||
" l: {\n",
|
||
" 1: {\n",
|
||
" c: df_crop_details[\n",
|
||
" (df_crop_details['cropName'] == c) &\n",
|
||
" (df_crop_details['cropLandType'] == l) &\n",
|
||
" ((df_crop_details['season'] == \"单季\") |\n",
|
||
" (df_crop_details['season'] == \"第一季\"))\n",
|
||
" ]['unitYield'].values[0] if df_crop_details[\n",
|
||
" (df_crop_details['cropName'] == c) &\n",
|
||
" (df_crop_details['cropLandType'] == l) &\n",
|
||
" ((df_crop_details['season'] == \"单季\") |\n",
|
||
" (df_crop_details['season'] == \"第一季\"))\n",
|
||
" ]['unitYield'].values.size > 0 else 0\n",
|
||
" for c in CropName\n",
|
||
" },\n",
|
||
" 2: {\n",
|
||
" c: df_crop_details[\n",
|
||
" (df_crop_details['cropName'] == c) &\n",
|
||
" (df_crop_details['cropLandType'] == l) &\n",
|
||
" (df_crop_details['season'] == \"第二季\")\n",
|
||
" ]['unitYield'].values[0] if df_crop_details[\n",
|
||
" (df_crop_details['cropName'] == c) &\n",
|
||
" (df_crop_details['cropLandType'] == l) &\n",
|
||
" (df_crop_details['season'] == \"第二季\")][\n",
|
||
" 'unitYield'].values.size > 0 else 0\n",
|
||
" for c in CropName\n",
|
||
" }\n",
|
||
" }\n",
|
||
" for l in LandType.values()\n",
|
||
"}\n",
|
||
"\n",
|
||
"# unit_yield_lsc"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 15
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "99dbe4ca6c54db0b",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.134526Z",
|
||
"start_time": "2024-09-08T08:06:04.947314Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 亩利润\n",
|
||
"unit_profit_lsc = {\n",
|
||
" l: {\n",
|
||
" 1: {\n",
|
||
" c: df_crop_details[\n",
|
||
" (df_crop_details['cropName'] == c) &\n",
|
||
" (df_crop_details['cropLandType'] == l) &\n",
|
||
" ((df_crop_details['season'] == \"单季\") |\n",
|
||
" (df_crop_details['season'] == \"第一季\"))\n",
|
||
" ]['unitProfit'].values[0] if df_crop_details[\n",
|
||
" (df_crop_details['cropName'] == c) &\n",
|
||
" (df_crop_details['cropLandType'] == l) &\n",
|
||
" ((df_crop_details['season'] == \"单季\") |\n",
|
||
" (df_crop_details['season'] == \"第一季\"))\n",
|
||
" ]['unitProfit'].values.size > 0 else 0\n",
|
||
" for c in CropName\n",
|
||
" },\n",
|
||
" 2: {\n",
|
||
" c: df_crop_details[\n",
|
||
" (df_crop_details['cropName'] == c) &\n",
|
||
" (df_crop_details['cropLandType'] == l) &\n",
|
||
" (df_crop_details['season'] == \"第二季\")\n",
|
||
" ]['unitProfit'].values[0] if df_crop_details[\n",
|
||
" (df_crop_details['cropName'] == c) &\n",
|
||
" (df_crop_details['cropLandType'] == l) &\n",
|
||
" (df_crop_details['season'] == \"第二季\")\n",
|
||
" ]['unitProfit'].values.size > 0 else 0\n",
|
||
" for c in CropName\n",
|
||
" }\n",
|
||
" }\n",
|
||
" for l in LandType.values()\n",
|
||
"}\n",
|
||
"\n",
|
||
"# unit_profit_lsc"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 16
|
||
},
|
||
{
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.323451Z",
|
||
"start_time": "2024-09-08T08:06:05.135577Z"
|
||
}
|
||
},
|
||
"cell_type": "code",
|
||
"source": [
|
||
"# 亩成本\n",
|
||
"unit_cost_lsc = {\n",
|
||
" l: {\n",
|
||
" 1: {\n",
|
||
" c: df_crop_details[\n",
|
||
" (df_crop_details['cropName'] == c) &\n",
|
||
" (df_crop_details['cropLandType'] == l) &\n",
|
||
" ((df_crop_details['season'] == \"单季\") |\n",
|
||
" (df_crop_details['season'] == \"第一季\"))\n",
|
||
" ]['cost'].values[0] if df_crop_details[\n",
|
||
" (df_crop_details['cropName'] == c) &\n",
|
||
" (df_crop_details['cropLandType'] == l) &\n",
|
||
" ((df_crop_details['season'] == \"单季\") |\n",
|
||
" (df_crop_details['season'] == \"第一季\"))\n",
|
||
" ]['cost'].values.size > 0 else 0\n",
|
||
" for c in CropName\n",
|
||
" },\n",
|
||
" 2: {\n",
|
||
" c: df_crop_details[\n",
|
||
" (df_crop_details['cropName'] == c) &\n",
|
||
" (df_crop_details['cropLandType'] == l) &\n",
|
||
" (df_crop_details['season'] == \"第二季\")\n",
|
||
" ]['cost'].values[0] if df_crop_details[\n",
|
||
" (df_crop_details['cropName'] == c) &\n",
|
||
" (df_crop_details['cropLandType'] == l) &\n",
|
||
" (df_crop_details['season'] == \"第二季\")\n",
|
||
" ]['cost'].values.size > 0 else 0\n",
|
||
" for c in CropName\n",
|
||
" }\n",
|
||
" }\n",
|
||
" for l in LandType.values()\n",
|
||
"}\n",
|
||
"# unit_cost_lsc"
|
||
],
|
||
"id": "1c8ad73eaea8be7a",
|
||
"outputs": [],
|
||
"execution_count": 17
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "13f2ccc7aa215d4c",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.338917Z",
|
||
"start_time": "2024-09-08T08:06:05.324337Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 每种作物的总需求\n",
|
||
"crop_demand = {\n",
|
||
" c: 0\n",
|
||
" for c in CropName\n",
|
||
"}\n",
|
||
"\n",
|
||
"# 这里需要另一张表\n",
|
||
"# 代码独立出来移到上面去了\n",
|
||
"\n",
|
||
"for line in df_crop_planting.values:\n",
|
||
" # 面积*该土地类型的亩产量 面积 地块类型字典 地块类型 季节 作物名称\n",
|
||
" crop_demand[line[2]] += line[4] * unit_yield_lsc[LandType[line[0][0]]][SeasonDict[line[5]]][line[2]]\n",
|
||
"\n",
|
||
"# crop_demand"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 18
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "5d09872708d40da4",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.386035Z",
|
||
"start_time": "2024-09-08T08:06:05.339917Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 准备开搞\n",
|
||
"from pulp import LpMaximize, LpProblem, LpVariable, lpSum, value, PULP_CBC_CMD, LpContinuous, LpBinary\n",
|
||
"import pulp"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 19
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "b6eff87c7a762769",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.401548Z",
|
||
"start_time": "2024-09-08T08:06:05.387014Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"model = LpProblem(\"Crop_Planting_Optimization_with_Specific_Rules\", LpMaximize)"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 20
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "7bfdb9425c3e9683",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.417576Z",
|
||
"start_time": "2024-09-08T08:06:05.402547Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# %timeit X = LpVariable.dicts(\"X\", (CropName, LandName, years, seasons), lowBound=0, cat=LpContinuous)\n",
|
||
"# %timeit XX = LpVariable.dicts(\"X\", [(c, l, y, s) for c in CropName for l in LandName for y in years for s in seasons],lowBound=0, cat=LpContinuous)"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 21
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "4cb129e20385a7ee",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.465493Z",
|
||
"start_time": "2024-09-08T08:06:05.418576Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"\n",
|
||
"t = []\n",
|
||
"for y in years:\n",
|
||
" for l in LandName:\n",
|
||
" for s in SeasonNum:\n",
|
||
" if l[0] in [\"A\", \"B\", \"C\"] and s == 1:\n",
|
||
" for c in CropType[\"全粮食（除了水稻）\"]:\n",
|
||
" t.append((c, l, y, s))\n",
|
||
" if l[0] == \"D\":\n",
|
||
" if s == 1:\n",
|
||
" for c in list(CropType[\"普通蔬菜\"]) + [\"水稻\"]:\n",
|
||
" t.append((c, l, y, s))\n",
|
||
" if s == 2:\n",
|
||
" for c in CropType[\"特殊蔬菜\"]:\n",
|
||
" t.append((c, l, y, s))\n",
|
||
" if l[0] == \"E\":\n",
|
||
" if s == 1:\n",
|
||
" for c in list(CropType[\"普通蔬菜\"]):\n",
|
||
" t.append((c, l, y, s))\n",
|
||
" if s == 2:\n",
|
||
" for c in CropType[\"食用菌\"]:\n",
|
||
" t.append((c, l, y, s))\n",
|
||
" if l[0] == \"F\":\n",
|
||
" for s in SeasonNum:\n",
|
||
" for c in CropType[\"普通蔬菜\"]:\n",
|
||
" t.append((c, l, y, s))\n",
|
||
"\n",
|
||
"\n",
|
||
"# X"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 22
|
||
},
|
||
{
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.496947Z",
|
||
"start_time": "2024-09-08T08:06:05.466594Z"
|
||
}
|
||
},
|
||
"cell_type": "code",
|
||
"source": [
|
||
"# 地块l在y年的s季的种植c的量\n",
|
||
"X = LpVariable.dicts(\"X\", t,\n",
|
||
" lowBound=0,\n",
|
||
" cat=LpContinuous\n",
|
||
" )"
|
||
],
|
||
"id": "ca23e006a68f2993",
|
||
"outputs": [],
|
||
"execution_count": 23
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "9dab983ca7b3b607",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.528472Z",
|
||
"start_time": "2024-09-08T08:06:05.497934Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 地块l在y年的s季是否种植了c\n",
|
||
"# Y = LpVariable.dicts(\"crop_plant_bool\", (CropName, LandName, years, SeasonType), cat=LpBinary)\n",
|
||
"# Y = LpVariable.dicts(\"Y\", [(c, l, y, s) for c in CropName for l in LandName for y in years for s in SeasonNum],\n",
|
||
"# cat=LpBinary)\n",
|
||
"Y = LpVariable.dicts(\"Y\", t, cat=LpBinary)\n",
|
||
"\n",
|
||
"# Y"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 24
|
||
},
|
||
{
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.543616Z",
|
||
"start_time": "2024-09-08T08:06:05.530473Z"
|
||
}
|
||
},
|
||
"cell_type": "code",
|
||
"source": [
|
||
"# def is_feasible(c, l, y, s):\n",
|
||
"# if l[0] in [\"A\", \"B\", \"C\"] and (c not in CropType[\"全粮食（除了水稻）\"]):\n",
|
||
"# return False\n",
|
||
"# if l[0] == \"D\" and (c not in CropType[\"全蔬菜\"] and c!=\"水稻\"):\n",
|
||
"# return False\n",
|
||
"# if l[0] == 'E' and s==2 and (c not in CropType[\"食用菌\"]):\n",
|
||
"# return False\n",
|
||
"# if l[0] == 'F' and (c not in CropType[\"全蔬菜\"]):\n",
|
||
"# return False\n",
|
||
"# return True"
|
||
],
|
||
"id": "28c6686374b8ed76",
|
||
"outputs": [],
|
||
"execution_count": 25
|
||
},
|
||
{
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.559071Z",
|
||
"start_time": "2024-09-08T08:06:05.545501Z"
|
||
}
|
||
},
|
||
"cell_type": "code",
|
||
"source": [
|
||
"tt = set(t)\n",
|
||
"\n",
|
||
"\n",
|
||
"def is_feasible(c, l, y, s):\n",
|
||
" return (c, l, y, s) in tt"
|
||
],
|
||
"id": "bf16c7f3ea10152f",
|
||
"outputs": [],
|
||
"execution_count": 26
|
||
},
|
||
{
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.745561Z",
|
||
"start_time": "2024-09-08T08:06:05.560052Z"
|
||
}
|
||
},
|
||
"cell_type": "code",
|
||
"source": [
|
||
"# 确保与X进行01约束\n",
|
||
"# for c in CropName:\n",
|
||
"# for l in LandName:\n",
|
||
"# for y in years:\n",
|
||
"# for s in SeasonNum:\n",
|
||
"# # if 1:\n",
|
||
"# if is_feasible(c, l, y, s):\n",
|
||
"# # 如果种植了作物，则种植面积大于0\n",
|
||
"# model += X[c, l, y, s] <= 100 * Y[c, l, y, s], f\"PlantingConstraint1_{c}_{l}_{y}_{s}\"\n",
|
||
"# # 如果未种植作物，则种植面积为0\n",
|
||
"# model += X[c, l, y, s] >= 0.001 * Y[c, l, y, s], f\"PlantingConstraint2_{c}_{l}_{y}_{s}\"\n",
|
||
"\n",
|
||
"\n",
|
||
"min_area = 0.2\n",
|
||
"for c in CropName:\n",
|
||
" for l in LandName:\n",
|
||
" for y in years:\n",
|
||
" for s in SeasonNum:\n",
|
||
" if is_feasible(c, l, y, s):\n",
|
||
" model += X[c, l, y, s] <= LandArea[l] * Y[c, l, y, s], f\"LandAreaConstraint_{c}_{l}_{y}_{s}\"\n",
|
||
" model += X[c, l, y, s] >= min_area * (\n",
|
||
" 1 if Y[c, l, y, s] else 0), f\"MinAreaConstraint_{c}_{l}_{y}_{s}\""
|
||
],
|
||
"id": "603175d36624172f",
|
||
"outputs": [],
|
||
"execution_count": 27
|
||
},
|
||
{
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.839721Z",
|
||
"start_time": "2024-09-08T08:06:05.748539Z"
|
||
}
|
||
},
|
||
"cell_type": "code",
|
||
"source": [
|
||
"# 目标函数：总利润 = sum(x[i]*profit[c,l,s])\n",
|
||
"# model += lpSum( # 地块类型 季节 作物类型\n",
|
||
"# X[c, l, y, s] * unit_profit_lsc[LandType[l[0]]][s][c]\n",
|
||
"# for c, l, y, s in t\n",
|
||
"# ), \"total_profit\"\n",
|
||
"\n",
|
||
"# 定义超过需求的变量\n",
|
||
"# Excess = LpVariable.dicts(\"Excess\", [(c, y) for c in CropName for y in years], lowBound=0)\n",
|
||
"\n",
|
||
"# 初始化总目标函数的表达式\n",
|
||
"total_profit = lpSum([]) # 空的lpSum用于初始化\n",
|
||
"\n",
|
||
"# 累积每个作物、每个年份的利润和惩罚\n",
|
||
"for c in CropName:\n",
|
||
" for y in years:\n",
|
||
" # 计算利润部分\n",
|
||
" profit_term = lpSum(\n",
|
||
" X[c, l, y, s] * unit_profit_lsc[LandType[l[0]]][s][c]\n",
|
||
" for l in LandName for s in SeasonNum if is_feasible(c, l, y, s)\n",
|
||
" )\n",
|
||
" \n",
|
||
" # 惩罚超过需求的部分：将每个地块、每个季节的单位成本与种植量相乘\n",
|
||
" penalty_term = lpSum(\n",
|
||
" X[c, l, y, s] * unit_cost_lsc[LandType[l[0]]][s][c]\n",
|
||
" for l in LandName for s in SeasonNum if is_feasible(c, l, y, s)\n",
|
||
" ) - crop_demand[c] * lpSum(\n",
|
||
" unit_cost_lsc[LandType[l[0]]][s][c]\n",
|
||
" for l in LandName for s in SeasonNum if is_feasible(c, l, y, s)\n",
|
||
" )\n",
|
||
" \n",
|
||
" # 累积到总目标函数中\n",
|
||
" total_profit += profit_term - penalty_term\n",
|
||
"\n",
|
||
"# 一次性将目标函数添加到模型中\n",
|
||
"model += total_profit, \"total_profit\"\n",
|
||
"# # 累积每个作物、每个年份的利润和惩罚\n",
|
||
"# for c in CropName:\n",
|
||
"# for y in years:\n",
|
||
"# # 计算利润部分，单位利润减去单位成本\n",
|
||
"# profit_term = lpSum(\n",
|
||
"# X[c, l, y, s] * (unit_profit_lsc[LandType[l[0]]][s][c])\n",
|
||
"# for l in LandName for s in SeasonNum if is_feasible(c, l, y, s)\n",
|
||
"# )\n",
|
||
"# \n",
|
||
"# # 惩罚超过需求的部分\n",
|
||
"# penalty_term = Excess[c, y] * lpSum(unit_cost_lsc[LandType[l[0]]][s][c] for l in LandName for s in SeasonNum if\n",
|
||
"# is_feasible(c, l, y, s))\n",
|
||
"# \n",
|
||
"# # 累积到总目标函数中\n",
|
||
"# total_profit += profit_term - penalty_term\n",
|
||
"# \n",
|
||
"# # 一次性将目标函数添加到模型中\n",
|
||
"# model += total_profit, \"total_profit\"\n",
|
||
"# \n",
|
||
"# \n",
|
||
"# # 目标函数：利润部分和惩罚部分\n",
|
||
"# for c in CropName:\n",
|
||
"# for y in years:\n",
|
||
"# # 计算利润部分，单位利润减去单位成本\n",
|
||
"# profit_term = lpSum(\n",
|
||
"# X[c, l, y, s] * (unit_profit_lsc[LandType[l[0]]][s][c])\n",
|
||
"# for l in LandName for s in SeasonNum if is_feasible(c, l, y, s)\n",
|
||
"# )\n",
|
||
"# \n",
|
||
"# # 惩罚超过需求的部分\n",
|
||
"# penalty_term = Excess[c, y] * lpSum(unit_cost_lsc[LandType[l[0]]][s][c] for l in LandName for s in SeasonNum if\n",
|
||
"# is_feasible(c, l, y, s)) # penalty_cost_per_unit 是每单位超过需求的惩罚成本\n",
|
||
"# \n",
|
||
"# # 添加到目标函数中\n",
|
||
"# model += profit_term - penalty_term, f\"total_profit_{c}_{y}\"\n"
|
||
],
|
||
"id": "4505b0f2fe6bf93d",
|
||
"outputs": [],
|
||
"execution_count": 28
|
||
},
|
||
{
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.855563Z",
|
||
"start_time": "2024-09-08T08:06:05.840637Z"
|
||
}
|
||
},
|
||
"cell_type": "code",
|
||
"source": [
|
||
"# # 添加需求约束\n",
|
||
"# for c in CropName:\n",
|
||
"# for y in years:\n",
|
||
"# model += lpSum(\n",
|
||
"# X[c, l, y, s] for l in LandName for s in SeasonNum if is_feasible(c, l, y, s)\n",
|
||
"# ) <= crop_demand[c] + Excess[c, y], f\"demand_constraint_{c}_{y}\""
|
||
],
|
||
"id": "a2fe5874538f1596",
|
||
"outputs": [],
|
||
"execution_count": 29
|
||
},
|
||
{
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.870378Z",
|
||
"start_time": "2024-09-08T08:06:05.856571Z"
|
||
}
|
||
},
|
||
"cell_type": "code",
|
||
"source": [
|
||
"# 约束，启动！\n",
|
||
"# 条件1：每年的所有季节中每种作物的种植量必须小于需求量\n",
|
||
"# for c in CropName:\n",
|
||
"# for y in years:\n",
|
||
"# model += lpSum(X[c, l, y, s] * unit_yield_lsc[LandType[l[0]]][s][c] for l in LandName for s in SeasonNum if\n",
|
||
"# is_feasible(c, l, y, s)) <= \\\n",
|
||
"# crop_demand[c], f\"CropDemandConstraint_{c}_{y}\""
|
||
],
|
||
"id": "1f185c0856fbcf27",
|
||
"outputs": [],
|
||
"execution_count": 30
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "7e13de7700d38f19",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.901626Z",
|
||
"start_time": "2024-09-08T08:06:05.871321Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 条件2： 每年每季每块地的种植面积不能超过该地块的总面积\n",
|
||
"for l in LandName:\n",
|
||
" for y in years:\n",
|
||
" for s in SeasonNum:\n",
|
||
" model += lpSum(X[c, l, y, s] for c in CropName if is_feasible(c, l, y, s)) <= LandArea[\n",
|
||
" l], f\"LandAreaConstraint_{l}_{y}_{s}\""
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 31
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "1f6ac4323825f28b",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.917541Z",
|
||
"start_time": "2024-09-08T08:06:05.902521Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 条件3： 平旱地、梯田和山坡地每年适宜单季种植[粮食类作物（水稻除外）]。因为除了[粮食类作物（水稻除外）]以外的作物似乎已经被上面的产量和利润的0给约束了，所以大概不需要做限制了吧\n",
|
||
"for l in LandName:\n",
|
||
" if l[0] in [\"A\", \"B\", \"C\"]: # [\"平旱地\", \"梯田\", \"山坡地\"]\n",
|
||
" for y in years:\n",
|
||
" model += lpSum(\n",
|
||
" Y[c, l, y, 2] for c in CropName if is_feasible(c, l, y, 2)) == 0, f\"SingleSeasonConstraint1_{y}_{l}\"\n",
|
||
" # model += lpSum(Y[c, l, y, 1] for c in CropType[\"全粮食（除了水稻）\"] if is_feasible(c, l, y, 1)) >= 1, f\"SingleSeasonConstraint2_{y}_{l}\"\n"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 32
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "c0e9b6b37092f0e7",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.933622Z",
|
||
"start_time": "2024-09-08T08:06:05.918440Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 条件4： 水浇地每年可以单季种植水稻或两季种植蔬菜作物。\n",
|
||
"for l in LandName:\n",
|
||
" if l[0] == \"D\":\n",
|
||
" for y in years:\n",
|
||
" for c in CropType[\"全蔬菜\"]:\n",
|
||
" if is_feasible(c,l,y,2):\n",
|
||
" model += Y[\"水稻\", l, y, 1] + Y[c, l, y, 2] <= 1, f\"IrrigatedConstraint0_{l}_{y}_{c}\"\n",
|
||
"\n",
|
||
"# for l in LandName:\n",
|
||
"# if l[0] in [\"D\"]:\n",
|
||
"# model += lpSum(Y[c, l, y, s] for c in CropType[\"全粮食（除了水稻）\"] for y in years for s in SeasonNum) == 0, f\"IrrigatedConstraint3_{l}\"\n",
|
||
"\n",
|
||
"# if l[0] != \"D\": # [\"水浇地\"]\n",
|
||
"# continue\n",
|
||
"# for y in years:\n",
|
||
"# model += (\n",
|
||
"# (lpSum(Y[\"水稻\", l, y, \"单季\"]) != 0) and\n",
|
||
"# (lpSum(\n",
|
||
"# Y[c, l, y, s]\n",
|
||
"# for c in CropType['全蔬菜']\n",
|
||
"# for s in [\"第一季\", \"第二季\"]\n",
|
||
"# ) == 0)\n",
|
||
"# ) or (\n",
|
||
"# (lpSum(Y[\"水稻\", l, y, \"单季\"]) == 0) and\n",
|
||
"# (lpSum(\n",
|
||
"# Y[c, l, y, s]\n",
|
||
"# for c in CropType['全蔬菜']\n",
|
||
"# for s in [\"第一季\", \"第二季\"]\n",
|
||
"# ) != 0)\n",
|
||
"# ), f\"irrigatedConstraint_{l}_{y}\"\n",
|
||
"\n"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 33
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "8f8da502641f32fa",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.948896Z",
|
||
"start_time": "2024-09-08T08:06:05.934627Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 条件5： 若在某块水浇地种植两季蔬菜，第一季可种植多种蔬菜（大白菜、白萝卜和红萝卜除外）；第二季只能种植大白菜、白萝卜和红萝卜中的一种（便于管理）。\n",
|
||
"# 感觉没必要，因为前面有field和profit为0的惩罚\n",
|
||
"# for l in LandName:\n",
|
||
"# if l[0] != \"D\": # [\"水浇地\"]\n",
|
||
"# continue\n",
|
||
"# for c in CropType['全蔬菜']:\n",
|
||
"# for y in years:\n",
|
||
"# if c not in CropType['特殊蔬菜']:\n",
|
||
"# if is_feasible(c, l, y, 2):\n",
|
||
"# model += Y[c, l, y, 2] == 0, f\"SpecialCropConstraint_{l}_{c}_{y}_第二季\"\n",
|
||
"# else:\n",
|
||
"# if is_feasible(c, l, y, 1):\n",
|
||
"# model += Y[c, l, y, 1] == 0, f\"SpecialCropConstraint_{l}_{c}_{y}_第一季\"\n",
|
||
" # if c not in CropType['特殊蔬菜']:\n",
|
||
" # model += lpSum(Y[c, l, y, 2] for y in years) == 0, f\"SpecialCropConstraint_{l}_{c}_第二季\"\n",
|
||
" # else:\n",
|
||
" # model += lpSum(Y[c, l, y, 1] for y in years) == 0, f\"SpecialCropConstraint_{l}_{c}_第一季\"\n",
|
||
"\n",
|
||
"# for l in LandName:\n",
|
||
"# if l[0] != \"D\": # [\"水浇地\"]\n",
|
||
"# continue\n",
|
||
"# for y in years:\n",
|
||
"# model += lpSum(Y[\"水稻\", l, y, s] for s in [\"第一季\", \"第二季\"]) == 0, f\"SpecialCropConstraint1_{l}_{y}\"\n",
|
||
"# model += (lpSum(Y[c, l, y, s] for s in [\"第一季\", \"第二季\"] for c in CropType['全蔬菜']) >= 0) and (\n",
|
||
"# lpSum(Y[\"水稻\", l, y, \"单季\"]) == 0), f\"SpecialCropConstraint2_{l}_{y}\""
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 34
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "df2d7e8918382a0",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.964928Z",
|
||
"start_time": "2024-09-08T08:06:05.950903Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 条件6： 根据季节性要求，大白菜、白萝卜和红萝卜只能在水浇地的第二季种植。\n",
|
||
"# for c in CropType[\"特殊蔬菜\"]:\n",
|
||
"# # for l in LandName:\n",
|
||
"# # if l[0] != \"D\": # [\"水浇地\"]\n",
|
||
"# # model += lpSum(Y[c, l, y, s] for y in years for s in SeasonNum for l in LandName if\n",
|
||
"# # l[0] != \"D\") == 0, f\"SpecialCropConstraint_{c}\"\n",
|
||
"# for l in LandName:\n",
|
||
"# if l[0] != \"D\": # [\"水浇地\"]\n",
|
||
"# for y in years:\n",
|
||
"# for s in SeasonNum:\n",
|
||
"# if is_feasible(c, l, y, s):\n",
|
||
"# model += Y[c, l, y, s] == 0, f\"SpecialCropConstraint_{c}_{l}_{y}_{s}\"\n"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 35
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "1d402d3df13f6732",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.980443Z",
|
||
"start_time": "2024-09-08T08:06:05.965932Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 条件7： 大棚没有单季。普通大棚每年种植两季作物，第一季可种植多种蔬菜（大白菜、白萝卜和红萝卜除外），第二季只能种植食用菌。\n",
|
||
"# for l in LandName:\n",
|
||
"# for y in years:\n",
|
||
"# for c in CropName:\n",
|
||
"# if l[0] in [\"E\", \"F\"]: # 大棚\n",
|
||
"# model += lpSum(Y[c, l, y, \"单季\"]) == 0, f\"SpecialCropConstraint_{c}_{y}_{l}\"\n",
|
||
"\n",
|
||
"# for l in LandName:\n",
|
||
"# if l[0] == \"E\": # [\"普通大棚\"]\n",
|
||
"# # model += lpSum(Y[c, l, y, 1] for c in CropType[\"特殊蔬菜\"] for y in years) == 0, f\"SpecialCropConstraint_{l}\"\n",
|
||
"# # model += lpSum(Y[c, l, y, 2] for c in CropName if c not in CropType[\"食用菌\"] for y in\n",
|
||
"# # years) == 0, f\"SpecialCropConstraint1__{l}\"\n",
|
||
"# for y in years:\n",
|
||
"# for c in CropType[\"特殊蔬菜\"]:\n",
|
||
"# if is_feasible(c, l, y, 1):\n",
|
||
"# model += Y[c, l, y, 1] == 0, f\"SpecialCropConstraint1_{l}_{y}_{c}\"\n",
|
||
"# for c in CropName:\n",
|
||
"# if c not in CropType[\"食用菌\"]:\n",
|
||
"# if is_feasible(c, l, y, 2):\n",
|
||
"# model += Y[c, l, y, 2] == 0, f\"SpecialCropConstraint2_{l}_{y}_{c}\"\n"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 36
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "930ab378623172ec",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:05.996056Z",
|
||
"start_time": "2024-09-08T08:06:05.982444Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 条件8： 食用菌类只能在普通大棚第二季的时候种植\n",
|
||
"# for c in CropType[\"食用菌\"]:\n",
|
||
"# # model += lpSum(Y[c, l, y, s] for y in years for l in LandName if l[0] != \"E\" for s in\n",
|
||
"# # SeasonNum) == 0, f\"SpecialCropConstraint2_{c}\"\n",
|
||
"# # model += lpSum(Y[c, l, y, 1] for l in LandName if l[0] == \"E\" for y in years) == 0, f\"SpecialCropConstraint3_{c}\"\n",
|
||
"# for y in years:\n",
|
||
"# for s in SeasonNum:\n",
|
||
"# model += lpSum(Y[c, l, y, s] for l in LandName if l[0] != \"E\" if\n",
|
||
"# is_feasible(c, l, y, s)) == 0, f\"SpecialCropConstraint2_{y}_{c}_{s}\"\n",
|
||
"# model += lpSum(Y[c, l, y, 1] for l in LandName if l[0] == \"E\" if\n",
|
||
"# is_feasible(c, l, y, 1)) == 0, f\"SpecialCropConstraint3_{y}_{c}\"\n",
|
||
"# for l in LandName:\n",
|
||
"# if l[0] != \"E\":\n",
|
||
"# for y in years:\n",
|
||
"# for s in SeasonType:\n",
|
||
"# model += lpSum(Y[c, l, y, s] for c in CropType[\"食用菌\"]) == 0, f\"SpecialCropConstraint1_{l}_{y}_{s}\"\n",
|
||
"# else:\n",
|
||
"# for y in years:\n",
|
||
"# model += lpSum(Y[c, l, y, \"第一季\"] for c in CropType[\"食用菌\"]) == 0, f\"SpecialCropConstraint2_{l}_{y}\"\n"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 37
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "bda8b4c538a097bc",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:06.011840Z",
|
||
"start_time": "2024-09-08T08:06:05.997121Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 条件9： 智慧大棚每年都可种植两季蔬菜（大白菜、白萝卜和红萝卜除外）\n",
|
||
"# for l in LandName:\n",
|
||
"# if l[0] == 'F':\n",
|
||
"# # model += lpSum(Y[c, l, y, s] for c in CropType[\"特殊蔬菜\"] for y in years for s in\n",
|
||
"# # SeasonNum) == 0, f\"SpecialCropConstraint4_{l}\"\n",
|
||
"# for y in years:\n",
|
||
"# for s in SeasonNum:\n",
|
||
"# model += lpSum(Y[c, l, y, s] for c in CropType[\"特殊蔬菜\"] if\n",
|
||
"# is_feasible(c, l, y, s)) == 0, f\"SpecialCropConstraint1_{l}_{y}_{s}\"\n",
|
||
" # for y in years:\n",
|
||
" # model += lpSum(Y[c, l, y, \"第一季\"] for c in CropType[\"全蔬菜\"] if\n",
|
||
" # c not in CropType[\"特殊蔬菜\"]) >= 0, f\"SpecialCropConstrain3_{l}_{y}\"\n",
|
||
" # model += lpSum(Y[c, l, y, \"第二季\"] for c in CropType[\"全蔬菜\"] if\n",
|
||
" # c not in CropType[\"特殊蔬菜\"]) >= 0, f\"SpecialCropConstraint4_{l}_{y}\"\n",
|
||
" # model += lpSum(\n",
|
||
" # Y[c, l, y, s] for c in CropType[\"特殊蔬菜\"] for s in SeasonType) == 0, f\"SpecialCropConstraint5_{l}_{y}\""
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 38
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "e8fb2dea1326a05b",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:06.043385Z",
|
||
"start_time": "2024-09-08T08:06:06.013345Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 条件10： 别种得太碎了，啥意思？ 每种作物不超过4块地吧。\n",
|
||
"for c in CropName:\n",
|
||
" for y in years:\n",
|
||
" for s in SeasonNum:\n",
|
||
" model += lpSum(\n",
|
||
" Y[c, l, y, s] for l in LandName if is_feasible(c, l, y, s)) <= 3, f\"MaxCropTypeConstraint_{c}_{y}_{s}\""
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 39
|
||
},
|
||
{
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:06.074027Z",
|
||
"start_time": "2024-09-08T08:06:06.044845Z"
|
||
}
|
||
},
|
||
"cell_type": "code",
|
||
"source": [
|
||
"# 条件11： 每块地最大种植作物种类数3\n",
|
||
"for l in LandName:\n",
|
||
" for y in years:\n",
|
||
" for s in SeasonNum:\n",
|
||
" model += lpSum(\n",
|
||
" Y[c, l, y, s] for c in CropName if is_feasible(c, l, y, s)) <= 3, f\"MaxLandConstraint_{l}_{y}_{s}\""
|
||
],
|
||
"id": "b8d9ae82090818ec",
|
||
"outputs": [],
|
||
"execution_count": 40
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "49f1490823c6af3c",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:06.152311Z",
|
||
"start_time": "2024-09-08T08:06:06.075083Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 条件12： 最小种植面积：0.2吧\n",
|
||
"for l in LandName:\n",
|
||
" for y in years:\n",
|
||
" for s in SeasonNum:\n",
|
||
" for c in CropName:\n",
|
||
" if is_feasible(c, l, y, s):\n",
|
||
" model += X[c, l, y, s] >= 0.2 * (\n",
|
||
" 1 if X[c, l, y, s] else 0), f\"MinCropAreaConstraint_{c}_{l}_{y}_{s}\""
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 41
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "bdf1048b83723f91",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:06.230187Z",
|
||
"start_time": "2024-09-08T08:06:06.153299Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 条件13 ： 最重要的，不能重复种植同一块地同一年同一季的同一作物\n",
|
||
"# 每种作物在同一地块（含大棚）都不能连续重茬种植\n",
|
||
"\n",
|
||
"for l in LandName:\n",
|
||
" for c in CropName:\n",
|
||
" for y in years[1:]: # 单季不能连年种植 同一年不能连续种植 跨年也不能连续种植\n",
|
||
" if is_feasible(c, l, y - 1, 1) and is_feasible(c, l, y - 1, 2):\n",
|
||
" model += Y[c, l, y - 1, 1] + Y[c, l, y - 1, 2] <= 1, f\"RepeatPlantingConstraint_{c}_{l}_{y}\"\n",
|
||
" model += Y[c, l, y - 1, 2] + Y[c, l, y, 1] <= 1, f\"RepeatPlantingConstraint1_{c}_{l}_{y}\"\n",
|
||
" if is_feasible(c, l, y - 1, 1) and is_feasible(c, l, y, 1):\n",
|
||
" if l[0] in [\"A\", \"B\", \"C\"]: # and c in CropType[\"全粮食\"]:\n",
|
||
" model += Y[c, l, y - 1, 1] + Y[c, l, y, 1] <= 1, f\"RepeatPlantingConstraint2_{c}_{l}_{y}\"\n",
|
||
" elif l[0] == \"D\" and c == \"水稻\":\n",
|
||
" model += Y[c, l, y - 1, 1] + Y[c, l, y, 1] <= 1, f\"RepeatPlantingConstraint3_{c}_{l}_{y}\"\n",
|
||
"\n",
|
||
" # if y < years[-1]:\n",
|
||
" # model += Y[c, l, y, \"单季\"] + Y[c, l, y + 1, \"单季\"] <= 1, f\"SingleSeasonConstraint1_{c}_{l}_{y}\"\n",
|
||
" # model += Y[c, l, y, \"第一季\"] + Y[c, l, y + 1, \"第二季\"] <= 1, f\"DoubleSeasonConstraint1_{c}_{l}_{y}\"\n",
|
||
" # model += Y[c, l, y, \"第一季\"] + Y[c, l, y, \"第二季\"] <= 1, f\"DoubleSeasonConstraint2_{c}_{l}_{y}\"\n"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 42
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "50da0aff9c68ae24",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:06.245603Z",
|
||
"start_time": "2024-09-08T08:06:06.230862Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 条件14：从 2023 年开始，要求每个地块（含大棚） 在三年内至少种植一次豆类作物\n",
|
||
"for l in LandName:\n",
|
||
" for y in years[2:]: # 只需循环到倒数第三年，因为要考虑三年周期\n",
|
||
" # 在 y-2 到 y+2 之间的三年内，至少有一次豆类作物\n",
|
||
" model += lpSum(X[c, l, y_r, s]\n",
|
||
" for c in CropType['豆类']\n",
|
||
" for y_r in range(y - 2, y + 1)\n",
|
||
" for s in SeasonNum if is_feasible(c, l, y_r, s)) >= LandArea[l], f\"BeanConstraint_{l}_{y}\"\n",
|
||
"\n",
|
||
"\n",
|
||
"\n",
|
||
"# 条件13： 最重要的，从 2023 年开始要求每个地块（含大棚） 的所有土地三年内至少种植一次豆类作物\n",
|
||
"# CropType['豆类']\n",
|
||
"# for l in LandName:\n",
|
||
"# for y in years[:-2]:\n",
|
||
"# model += lpSum(Y[c,l,y_r,s] for c in CropType['豆类'] for y_r in range(y,y+3) for s in SeasonType)>=1, f\"BeanConstraint_{l}_{y}\"\n",
|
||
"# model += lpSum(X[c, l, y_r, s] for c in CropType['豆类'] for y_r in range(y, y + 3) for s in SeasonType) >= \\\n",
|
||
"# LandArea[l], f\"BeanConstraint_{l}_{y}\"\n",
|
||
"# model += lpSum(Y[c, l, y_r, \"单季\"] for c in CropType['豆类'] for y_r in\n",
|
||
"# range(y - 2, y + 1)) >= 1, f\"BeanConstraint1_{l}_{y}\"\n",
|
||
"# model += lpSum(Y[c, l, y_r, s] for c in CropType['豆类'] for y_r in range(y - 2, y + 1) for s in\n",
|
||
"# [\"第一季\", \"第二季\"]) >= 1, f\"BeanConstraint2_{l}_{y}\""
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 43
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "d4792a1210afca3b",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:15.814241Z",
|
||
"start_time": "2024-09-08T08:06:06.246550Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"%%time\n",
|
||
"model.solve(\n",
|
||
" pulp.GUROBI_CMD(msg=True,\n",
|
||
" timeLimit=7200,\n",
|
||
" threads=20,\n",
|
||
" options=[(\"Method\", 2), (\"MIPGap\", 0.0001)]))\n",
|
||
"\n",
|
||
"# model.solve()\n",
|
||
"# 创建求解器实例\n",
|
||
"# solver = pulp.PULP_CBC_CMD(\n",
|
||
"# timeLimit=120,\n",
|
||
"# threads=4,\n",
|
||
"# )\n",
|
||
"# solver.solve(model)\n",
|
||
"# print(Y)\n",
|
||
"\n"
|
||
],
|
||
"outputs": [
|
||
{
|
||
"name": "stdout",
|
||
"output_type": "stream",
|
||
"text": [
|
||
"CPU times: total: 0 ns\n",
|
||
"Wall time: 9.56 s\n"
|
||
]
|
||
},
|
||
{
|
||
"data": {
|
||
"text/plain": [
|
||
"1"
|
||
]
|
||
},
|
||
"execution_count": 44,
|
||
"metadata": {},
|
||
"output_type": "execute_result"
|
||
}
|
||
],
|
||
"execution_count": 44
|
||
},
|
||
{
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:15.829436Z",
|
||
"start_time": "2024-09-08T08:06:15.815222Z"
|
||
}
|
||
},
|
||
"cell_type": "code",
|
||
"source": [
|
||
"# for c in CropName:\n",
|
||
"# for l in LandName:\n",
|
||
"# for y in years:\n",
|
||
"# for s in SeasonNum:\n",
|
||
"# print(f\"Y[({c}, {l}, {y}, {s})] = {pulp.value(Y[(c, l, y, s)])}\")"
|
||
],
|
||
"id": "c64b04effa808515",
|
||
"outputs": [],
|
||
"execution_count": 45
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "cb3f4e95111e3292",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:16.238091Z",
|
||
"start_time": "2024-09-08T08:06:15.830431Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"# 输出文件名，包含起始年份和结束年份\n",
|
||
"output_filename = f\"planting_plan_{years[0]}-{years[-1]}.xlsx\"\n",
|
||
"\n",
|
||
"with pd.ExcelWriter(output_filename) as writer:\n",
|
||
" for y in years:\n",
|
||
"\n",
|
||
" # 用于存储需要交换的DataFrame\n",
|
||
" sheets_to_swap = {}\n",
|
||
"\n",
|
||
" for s in SeasonNum:\n",
|
||
" # 创建一个空的 DataFrame，横轴为作物名称，纵轴为地块名称\n",
|
||
" result_table = pd.DataFrame(columns=CropName, index=LandName)\n",
|
||
"\n",
|
||
" # 填充表格数据\n",
|
||
" for l in LandName:\n",
|
||
" for c in CropName:\n",
|
||
" # 获取变量 X[c, l, y, s] 的值\n",
|
||
" if is_feasible(c, l, y, s):\n",
|
||
" crop_value = value(X[c, l, y, s])\n",
|
||
" if crop_value is not None and crop_value > 0:\n",
|
||
" result_table.at[l, c] = crop_value # 填入结果\n",
|
||
"\n",
|
||
" # 用 0 填充空值\n",
|
||
" result_table.fillna(0, inplace=True)\n",
|
||
" sheet_name = f\"{y}_Season_第{s}季\"\n",
|
||
" result_table.to_excel(writer, sheet_name=sheet_name)\n",
|
||
"\n",
|
||
"print(f\"结果已保存到文件：{output_filename}\")\n"
|
||
],
|
||
"outputs": [
|
||
{
|
||
"name": "stdout",
|
||
"output_type": "stream",
|
||
"text": [
|
||
"结果已保存到文件：planting_plan_2024-2030.xlsx\n"
|
||
]
|
||
}
|
||
],
|
||
"execution_count": 46
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"id": "ebbf10d3",
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:16.687917Z",
|
||
"start_time": "2024-09-08T08:06:16.239987Z"
|
||
}
|
||
},
|
||
"source": [
|
||
"import pandas as pd\n",
|
||
"\n",
|
||
"# 假设以下变量已经定义\n",
|
||
"# years, SeasonNum, CropName, LandName, X, value, output_filename\n",
|
||
"\n",
|
||
"with pd.ExcelWriter(output_filename) as writer:\n",
|
||
" for y in years:\n",
|
||
" # 用于存储需要交换的DataFrame\n",
|
||
" sheets_to_swap = {}\n",
|
||
"\n",
|
||
" for s in SeasonNum:\n",
|
||
" # 创建一个空的 DataFrame，横轴为作物名称，纵轴为地块名称\n",
|
||
" result_table = pd.DataFrame(columns=CropName, index=LandName)\n",
|
||
"\n",
|
||
" # 填充表格数据\n",
|
||
" for l in LandName:\n",
|
||
" for c in CropName:\n",
|
||
" # 获取变量 X[c, l, y, s] 的值\n",
|
||
" if is_feasible(c, l, y, s):\n",
|
||
" crop_value = value(X[c, l, y, s])\n",
|
||
" if crop_value is not None and crop_value > 0:\n",
|
||
" result_table.at[l, c] = crop_value # 填入结果\n",
|
||
"\n",
|
||
" # 用 0 填充空值\n",
|
||
" result_table.fillna(0, inplace=True)\n",
|
||
"\n",
|
||
" # 对DataFrame中的所有数值进行四舍五入，保留两位小数\n",
|
||
" result_table = result_table.round(2)\n",
|
||
"\n",
|
||
" sheet_name = f\"{y}_Season_第{s}季\"\n",
|
||
" result_table.to_excel(writer, sheet_name=sheet_name)\n"
|
||
],
|
||
"outputs": [],
|
||
"execution_count": 47
|
||
},
|
||
{
|
||
"metadata": {
|
||
"ExecuteTime": {
|
||
"end_time": "2024-09-08T08:06:16.703415Z",
|
||
"start_time": "2024-09-08T08:06:16.688918Z"
|
||
}
|
||
},
|
||
"cell_type": "code",
|
||
"source": "",
|
||
"id": "ab1fd16df70c4293",
|
||
"outputs": [],
|
||
"execution_count": 47
|
||
}
|
||
],
|
||
"metadata": {
|
||
"kernelspec": {
|
||
"display_name": "Python 3 (ipykernel)",
|
||
"language": "python",
|
||
"name": "python3"
|
||
},
|
||
"language_info": {
|
||
"codemirror_mode": {
|
||
"name": "ipython",
|
||
"version": 3
|
||
},
|
||
"file_extension": ".py",
|
||
"mimetype": "text/x-python",
|
||
"name": "python",
|
||
"nbconvert_exporter": "python",
|
||
"pygments_lexer": "ipython3",
|
||
"version": "3.9.17"
|
||
}
|
||
},
|
||
"nbformat": 4,
|
||
"nbformat_minor": 5
|
||
}
|